Halide 19.0.0
Halide compiler and libraries
Loading...
Searching...
No Matches
Halide::Internal::CodeGen_C Class Reference

This class emits C++ code equivalent to a halide Stmt. More...

#include <CodeGen_C.h>

Inheritance diagram for Halide::Internal::CodeGen_C:
Halide::Internal::IRPrinter Halide::Internal::IRVisitor Halide::Internal::CodeGen_GPU_C

Classes

struct  Allocation
 

Public Types

enum  OutputKind {
  CHeader , CPlusPlusHeader , CImplementation , CPlusPlusImplementation ,
  CExternDecl , CPlusPlusExternDecl , CPlusPlusFunctionInfoHeader
}
 

Public Member Functions

 CodeGen_C (std::ostream &dest, const Target &target, OutputKind output_kind=CImplementation, const std::string &include_guard="")
 Initialize a C code generator pointing at a particular output stream (e.g.
 
 ~CodeGen_C () override
 
void compile (const Module &module)
 Emit the declarations contained in the module as C code.
 
const Targetget_target () const
 The target we're generating code for.
 
- Public Member Functions inherited from Halide::Internal::IRPrinter
 IRPrinter (std::ostream &)
 Construct an IRPrinter pointed at a given output stream (e.g.
 
void print (const Expr &)
 emit an expression on the output stream
 
void print_no_parens (const Expr &)
 Emit an expression on the output stream without enclosing parens.
 
void print (const Stmt &)
 emit a statement on the output stream
 
void print_summary (const Stmt &)
 emit a statement summary on the output stream
 
void print_list (const std::vector< Expr > &exprs)
 emit a comma delimited list of exprs, without any leading or trailing punctuation.
 
- Public Member Functions inherited from Halide::Internal::IRVisitor
 IRVisitor ()=default
 
virtual ~IRVisitor ()=default
 

Static Public Member Functions

static void test ()
 
- Static Public Member Functions inherited from Halide::Internal::IRPrinter
static void test ()
 

Protected Types

enum class  IntegerSuffixStyle { PlainC = 0 , OpenCL = 1 , HLSL = 2 }
 
enum  AppendSpaceIfNeeded { DoNotAppendSpace , AppendSpace }
 

Protected Member Functions

void compile (const LoweredFunc &func, const MetadataNameMap &metadata_name_map)
 Emit a declaration.
 
void compile (const Buffer<> &buffer)
 
virtual Stmt preprocess_function_body (const Stmt &stmt)
 This is a hook that subclasses can use to transform a function body just before it is emitted – e.g., to transform the IR to code that is easier to recognize and emit.
 
std::string print_expr (const Expr &)
 Emit an expression as an assignment, then return the id of the resulting var.
 
std::string print_cast_expr (const Type &, const Expr &)
 Like print_expr, but cast the Expr to the given Type.
 
void print_stmt (const Stmt &)
 Emit a statement.
 
void create_assertion (const std::string &id_cond, const Expr &message)
 
void create_assertion (const Expr &cond, const Expr &message)
 
Expr scalarize_vector_reduce (const VectorReduce *op)
 
virtual std::string print_type (Type, AppendSpaceIfNeeded space_option=DoNotAppendSpace)
 Emit the C name for a halide type.
 
virtual std::string print_reinterpret (Type, const Expr &)
 Emit a statement to reinterpret an expression as another type.
 
virtual std::string print_name (const std::string &)
 Emit a version of a string that is a valid identifier in C (.
 
virtual void add_platform_prologue ()
 Add platform specific prologue.
 
virtual void add_vector_typedefs (const std::set< Type > &vector_types)
 Add typedefs for vector types.
 
virtual std::string print_extern_call (const Call *op)
 Bottleneck to allow customization of calls to generic Extern/PureExtern calls.
 
std::string print_scalarized_expr (const Expr &e)
 Convert a vector Expr into a series of scalar Exprs, then reassemble into vector of original type.
 
virtual std::string print_assignment (Type t, const std::string &rhs)
 Emit an SSA-style assignment, and set id to the freshly generated name.
 
void print_heap_free (const std::string &alloc_name)
 Emit free for the heap allocation.
 
bool is_header ()
 Return true if only generating an interface, which may be extern "C" or C++.
 
bool is_extern_decl ()
 Return true if only generating an interface, which may be extern "C" or C++.
 
bool is_header_or_extern_decl ()
 Return true if only generating an interface, which may be extern "C" or C++.
 
bool is_c_plus_plus_interface ()
 Return true if generating C++ linkage.
 
void open_scope ()
 Open a new C scope (i.e.
 
void close_scope (const std::string &comment)
 Close a C scope (i.e.
 
void forward_declare_type_if_needed (const Type &t)
 If the Type is a handle type, emit a forward-declaration for it if we haven't already.
 
void set_name_mangling_mode (NameMangling mode)
 
void visit (const Variable *) override
 
void visit (const IntImm *) override
 
void visit (const UIntImm *) override
 
void visit (const StringImm *) override
 
void visit (const FloatImm *) override
 
void visit (const Cast *) override
 
void visit (const Reinterpret *) override
 
void visit (const Add *) override
 
void visit (const Sub *) override
 
void visit (const Mul *) override
 
void visit (const Div *) override
 
void visit (const Mod *) override
 
void visit (const Max *) override
 
void visit (const Min *) override
 
void visit (const EQ *) override
 
void visit (const NE *) override
 
void visit (const LT *) override
 
void visit (const LE *) override
 
void visit (const GT *) override
 
void visit (const GE *) override
 
void visit (const And *) override
 
void visit (const Or *) override
 
void visit (const Not *) override
 
void visit (const Call *) override
 
void visit (const Select *) override
 
void visit (const Load *) override
 
void visit (const Store *) override
 
void visit (const Let *) override
 
void visit (const LetStmt *) override
 
void visit (const AssertStmt *) override
 
void visit (const ProducerConsumer *) override
 
void visit (const For *) override
 
void visit (const Ramp *) override
 
void visit (const Broadcast *) override
 
void visit (const Provide *) override
 
void visit (const Allocate *) override
 
void visit (const Free *) override
 
void visit (const Realize *) override
 
void visit (const IfThenElse *) override
 
void visit (const Evaluate *) override
 
void visit (const Shuffle *) override
 
void visit (const Prefetch *) override
 
void visit (const Fork *) override
 
void visit (const Acquire *) override
 
void visit (const Atomic *) override
 
void visit (const VectorReduce *) override
 
void visit_binop (Type t, const Expr &a, const Expr &b, const char *op)
 
void visit_relop (Type t, const Expr &a, const Expr &b, const char *scalar_op, const char *vector_op)
 
virtual bool is_stack_private_to_thread () const
 Some architectures have private memory for the call stack; this means a thread cannot hand pointers to stack memory to another thread.
 
void emit_argv_wrapper (const std::string &function_name, const std::vector< LoweredArgument > &args)
 
void emit_metadata_getter (const std::string &function_name, const std::vector< LoweredArgument > &args, const MetadataNameMap &metadata_name_map)
 
void emit_constexpr_function_info (const std::string &function_name, const std::vector< LoweredArgument > &args, const MetadataNameMap &metadata_name_map)
 
void emit_halide_free_helper (const std::string &alloc_name, const std::string &free_function)
 
void visit (const Block *) override
 
void visit (const HoistedStorage *) override
 
- Protected Member Functions inherited from Halide::Internal::IRPrinter
Indentation get_indent () const
 
void open ()
 Either emits "(" or "", depending on the value of implicit_parens.
 
void close ()
 Either emits ")" or "", depending on the value of implicit_parens.
 
void print_lets (const Let *let)
 A helper for printing a chain of lets with line breaks.
 
void print_braced_stmt (const Stmt &, int extra_indent=2)
 A helper for printing a braced statement.
 
void visit (const IntImm *) override
 
void visit (const UIntImm *) override
 
void visit (const FloatImm *) override
 
void visit (const StringImm *) override
 
void visit (const Cast *) override
 
void visit (const Reinterpret *) override
 
void visit (const Variable *) override
 
void visit (const Add *) override
 
void visit (const Sub *) override
 
void visit (const Mul *) override
 
void visit (const Div *) override
 
void visit (const Mod *) override
 
void visit (const Min *) override
 
void visit (const Max *) override
 
void visit (const EQ *) override
 
void visit (const NE *) override
 
void visit (const LT *) override
 
void visit (const LE *) override
 
void visit (const GT *) override
 
void visit (const GE *) override
 
void visit (const And *) override
 
void visit (const Or *) override
 
void visit (const Not *) override
 
void visit (const Select *) override
 
void visit (const Load *) override
 
void visit (const Ramp *) override
 
void visit (const Broadcast *) override
 
void visit (const Call *) override
 
void visit (const Let *) override
 
void visit (const LetStmt *) override
 
void visit (const AssertStmt *) override
 
void visit (const ProducerConsumer *) override
 
void visit (const For *) override
 
void visit (const Acquire *) override
 
void visit (const Store *) override
 
void visit (const Provide *) override
 
void visit (const Allocate *) override
 
void visit (const Free *) override
 
void visit (const Realize *) override
 
void visit (const Block *) override
 
void visit (const Fork *) override
 
void visit (const IfThenElse *) override
 
void visit (const Evaluate *) override
 
void visit (const Shuffle *) override
 
void visit (const VectorReduce *) override
 
void visit (const Prefetch *) override
 
void visit (const Atomic *) override
 
void visit (const HoistedStorage *) override
 
- Protected Member Functions inherited from Halide::Internal::IRVisitor

Static Protected Member Functions

template<typename T >
static std::string with_sep (const std::vector< T > &v, const std::string &sep)
 
template<typename T >
static std::string with_commas (const std::vector< T > &v)
 

Protected Attributes

IntegerSuffixStyle integer_suffix_style = IntegerSuffixStyle::PlainC
 How to emit 64-bit integer constants.
 
std::string id
 An ID for the most recently generated ssa variable.
 
Target target
 The target being generated for.
 
OutputKind output_kind
 Controls whether this instance is generating declarations or definitions and whether the interface us extern "C" or C++.
 
std::map< std::string, std::string > cache
 A cache of generated values in scope.
 
Scope< Allocationallocations
 Track the types of allocations to avoid unnecessary casts.
 
Scope heap_allocations
 Track which allocations actually went on the heap.
 
bool have_user_context
 True if there is a void * __user_context parameter in the arguments.
 
bool extern_c_open = false
 Track current calling convention scope.
 
bool uses_gpu_for_loops
 True if at least one gpu-based for loop is used.
 
std::set< const halide_handle_cplusplus_type * > forward_declared
 Track which handle types have been forward-declared already.
 
bool inside_atomic_mutex_node = false
 Are we inside an atomic node that uses mutex locks? This is used for detecting deadlocks from nested atomics.
 
bool emit_atomic_stores = false
 Emit atomic store instructions?
 
bool using_vector_typedefs = false
 true if add_vector_typedefs() has been called.
 
- Protected Attributes inherited from Halide::Internal::IRPrinter
std::ostream & stream
 The stream on which we're outputting.
 
int indent = 0
 The current indentation level, useful for pretty-printing statements.
 
bool implicit_parens = false
 Certain expressions do not need parens around them, e.g.
 
bool is_summary = false
 Print only a summary of a statement, with sub-statements replaced by ellipses (...).
 
Scope known_type
 The symbols whose types can be inferred from values printed already.
 

Detailed Description

This class emits C++ code equivalent to a halide Stmt.

It's mostly the same as an IRPrinter, but it's wrapped in a function definition, and some things are handled differently to be valid C++.

Definition at line 27 of file CodeGen_C.h.

Member Enumeration Documentation

◆ OutputKind

Enumerator
CHeader 
CPlusPlusHeader 
CImplementation 
CPlusPlusImplementation 
CExternDecl 
CPlusPlusExternDecl 
CPlusPlusFunctionInfoHeader 

Definition at line 29 of file CodeGen_C.h.

◆ IntegerSuffixStyle

Enumerator
PlainC 
OpenCL 
HLSL 

Definition at line 58 of file CodeGen_C.h.

◆ AppendSpaceIfNeeded

Enumerator
DoNotAppendSpace 
AppendSpace 

Definition at line 112 of file CodeGen_C.h.

Constructor & Destructor Documentation

◆ CodeGen_C()

Halide::Internal::CodeGen_C::CodeGen_C ( std::ostream & dest,
const Target & target,
OutputKind output_kind = CImplementation,
const std::string & include_guard = "" )

Initialize a C code generator pointing at a particular output stream (e.g.

a file, or std::cout)

◆ ~CodeGen_C()

Halide::Internal::CodeGen_C::~CodeGen_C ( )
override

Member Function Documentation

◆ compile() [1/3]

void Halide::Internal::CodeGen_C::compile ( const Module & module)

Emit the declarations contained in the module as C code.

◆ get_target()

const Target & Halide::Internal::CodeGen_C::get_target ( ) const
inline

The target we're generating code for.

Definition at line 51 of file CodeGen_C.h.

References target.

◆ test()

static void Halide::Internal::CodeGen_C::test ( )
static

◆ compile() [2/3]

void Halide::Internal::CodeGen_C::compile ( const LoweredFunc & func,
const MetadataNameMap & metadata_name_map )
protected

Emit a declaration.

◆ compile() [3/3]

void Halide::Internal::CodeGen_C::compile ( const Buffer<> & buffer)
protected

◆ preprocess_function_body()

virtual Stmt Halide::Internal::CodeGen_C::preprocess_function_body ( const Stmt & stmt)
protectedvirtual

This is a hook that subclasses can use to transform a function body just before it is emitted – e.g., to transform the IR to code that is easier to recognize and emit.

The default implementation simply returns the input unchanged.

This hook will always be called after the function declaration and opening brace is emitted, so in addition to (possibly) returning a modified Stmt, this function may also emit C++ code to the default stream if it wishes to add some prologue at the start of the function.

◆ print_expr()

std::string Halide::Internal::CodeGen_C::print_expr ( const Expr & )
protected

Emit an expression as an assignment, then return the id of the resulting var.

◆ print_cast_expr()

std::string Halide::Internal::CodeGen_C::print_cast_expr ( const Type & ,
const Expr &  )
protected

Like print_expr, but cast the Expr to the given Type.

◆ print_stmt()

void Halide::Internal::CodeGen_C::print_stmt ( const Stmt & )
protected

Emit a statement.

◆ create_assertion() [1/2]

void Halide::Internal::CodeGen_C::create_assertion ( const std::string & id_cond,
const Expr & message )
protected

◆ create_assertion() [2/2]

void Halide::Internal::CodeGen_C::create_assertion ( const Expr & cond,
const Expr & message )
protected

◆ scalarize_vector_reduce()

Expr Halide::Internal::CodeGen_C::scalarize_vector_reduce ( const VectorReduce * op)
protected

◆ print_type()

virtual std::string Halide::Internal::CodeGen_C::print_type ( Type ,
AppendSpaceIfNeeded space_option = DoNotAppendSpace )
protectedvirtual

Emit the C name for a halide type.

If space_option is AppendSpace, and there should be a space between the type and the next token, one is appended. (This allows both "int foo" and "Foo *foo" to be formatted correctly. Otherwise the latter is "Foo * foo".)

◆ print_reinterpret()

virtual std::string Halide::Internal::CodeGen_C::print_reinterpret ( Type ,
const Expr &  )
protectedvirtual

Emit a statement to reinterpret an expression as another type.

◆ print_name()

virtual std::string Halide::Internal::CodeGen_C::print_name ( const std::string & )
protectedvirtual

Emit a version of a string that is a valid identifier in C (.

is replaced with _)

◆ add_platform_prologue()

virtual void Halide::Internal::CodeGen_C::add_platform_prologue ( )
protectedvirtual

Add platform specific prologue.

◆ add_vector_typedefs()

virtual void Halide::Internal::CodeGen_C::add_vector_typedefs ( const std::set< Type > & vector_types)
protectedvirtual

Add typedefs for vector types.

Not needed for OpenCL, might use different syntax for other C-like languages.

◆ print_extern_call()

virtual std::string Halide::Internal::CodeGen_C::print_extern_call ( const Call * op)
protectedvirtual

Bottleneck to allow customization of calls to generic Extern/PureExtern calls.


◆ print_scalarized_expr()

std::string Halide::Internal::CodeGen_C::print_scalarized_expr ( const Expr & e)
protected

Convert a vector Expr into a series of scalar Exprs, then reassemble into vector of original type.


◆ print_assignment()

virtual std::string Halide::Internal::CodeGen_C::print_assignment ( Type t,
const std::string & rhs )
protectedvirtual

Emit an SSA-style assignment, and set id to the freshly generated name.

Return id.

◆ print_heap_free()

void Halide::Internal::CodeGen_C::print_heap_free ( const std::string & alloc_name)
protected

Emit free for the heap allocation.

◆ is_header()

bool Halide::Internal::CodeGen_C::is_header ( )
inlineprotected

Return true if only generating an interface, which may be extern "C" or C++.

Definition at line 150 of file CodeGen_C.h.

References CHeader, CPlusPlusFunctionInfoHeader, CPlusPlusHeader, and output_kind.

Referenced by is_header_or_extern_decl().

◆ is_extern_decl()

bool Halide::Internal::CodeGen_C::is_extern_decl ( )
inlineprotected

Return true if only generating an interface, which may be extern "C" or C++.

Definition at line 157 of file CodeGen_C.h.

References CExternDecl, CPlusPlusExternDecl, and output_kind.

Referenced by is_header_or_extern_decl().

◆ is_header_or_extern_decl()

bool Halide::Internal::CodeGen_C::is_header_or_extern_decl ( )
inlineprotected

Return true if only generating an interface, which may be extern "C" or C++.

Definition at line 163 of file CodeGen_C.h.

References is_extern_decl(), and is_header().

◆ is_c_plus_plus_interface()

bool Halide::Internal::CodeGen_C::is_c_plus_plus_interface ( )
inlineprotected

Return true if generating C++ linkage.

Definition at line 168 of file CodeGen_C.h.

References CPlusPlusExternDecl, CPlusPlusFunctionInfoHeader, CPlusPlusHeader, CPlusPlusImplementation, and output_kind.

◆ open_scope()

void Halide::Internal::CodeGen_C::open_scope ( )
protected

Open a new C scope (i.e.

throw in a brace, increase the indent)

◆ close_scope()

void Halide::Internal::CodeGen_C::close_scope ( const std::string & comment)
protected

Close a C scope (i.e.

throw in an end brace, decrease the indent)

◆ forward_declare_type_if_needed()

void Halide::Internal::CodeGen_C::forward_declare_type_if_needed ( const Type & t)
protected

If the Type is a handle type, emit a forward-declaration for it if we haven't already.

◆ set_name_mangling_mode()

void Halide::Internal::CodeGen_C::set_name_mangling_mode ( NameMangling mode)
protected

◆ visit() [1/48]

void Halide::Internal::CodeGen_C::visit ( const Variable * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [2/48]

void Halide::Internal::CodeGen_C::visit ( const IntImm * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [3/48]

void Halide::Internal::CodeGen_C::visit ( const UIntImm * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [4/48]

void Halide::Internal::CodeGen_C::visit ( const StringImm * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [5/48]

void Halide::Internal::CodeGen_C::visit ( const FloatImm * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [6/48]

void Halide::Internal::CodeGen_C::visit ( const Cast * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [7/48]

void Halide::Internal::CodeGen_C::visit ( const Reinterpret * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [8/48]

void Halide::Internal::CodeGen_C::visit ( const Add * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [9/48]

void Halide::Internal::CodeGen_C::visit ( const Sub * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [10/48]

void Halide::Internal::CodeGen_C::visit ( const Mul * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [11/48]

void Halide::Internal::CodeGen_C::visit ( const Div * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [12/48]

void Halide::Internal::CodeGen_C::visit ( const Mod * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [13/48]

void Halide::Internal::CodeGen_C::visit ( const Max * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [14/48]

void Halide::Internal::CodeGen_C::visit ( const Min * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [15/48]

void Halide::Internal::CodeGen_C::visit ( const EQ * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [16/48]

void Halide::Internal::CodeGen_C::visit ( const NE * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [17/48]

void Halide::Internal::CodeGen_C::visit ( const LT * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [18/48]

void Halide::Internal::CodeGen_C::visit ( const LE * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [19/48]

void Halide::Internal::CodeGen_C::visit ( const GT * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [20/48]

void Halide::Internal::CodeGen_C::visit ( const GE * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [21/48]

void Halide::Internal::CodeGen_C::visit ( const And * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [22/48]

void Halide::Internal::CodeGen_C::visit ( const Or * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [23/48]

void Halide::Internal::CodeGen_C::visit ( const Not * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [24/48]

void Halide::Internal::CodeGen_C::visit ( const Call * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [25/48]

void Halide::Internal::CodeGen_C::visit ( const Select * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [26/48]

void Halide::Internal::CodeGen_C::visit ( const Load * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [27/48]

void Halide::Internal::CodeGen_C::visit ( const Store * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [28/48]

void Halide::Internal::CodeGen_C::visit ( const Let * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [29/48]

void Halide::Internal::CodeGen_C::visit ( const LetStmt * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [30/48]

void Halide::Internal::CodeGen_C::visit ( const AssertStmt * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [31/48]

void Halide::Internal::CodeGen_C::visit ( const ProducerConsumer * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [32/48]

void Halide::Internal::CodeGen_C::visit ( const For * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [33/48]

void Halide::Internal::CodeGen_C::visit ( const Ramp * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [34/48]

void Halide::Internal::CodeGen_C::visit ( const Broadcast * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [35/48]

void Halide::Internal::CodeGen_C::visit ( const Provide * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [36/48]

void Halide::Internal::CodeGen_C::visit ( const Allocate * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [37/48]

void Halide::Internal::CodeGen_C::visit ( const Free * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [38/48]

void Halide::Internal::CodeGen_C::visit ( const Realize * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [39/48]

void Halide::Internal::CodeGen_C::visit ( const IfThenElse * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [40/48]

void Halide::Internal::CodeGen_C::visit ( const Evaluate * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [41/48]

void Halide::Internal::CodeGen_C::visit ( const Shuffle * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [42/48]

void Halide::Internal::CodeGen_C::visit ( const Prefetch * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [43/48]

void Halide::Internal::CodeGen_C::visit ( const Fork * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [44/48]

void Halide::Internal::CodeGen_C::visit ( const Acquire * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [45/48]

void Halide::Internal::CodeGen_C::visit ( const Atomic * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [46/48]

void Halide::Internal::CodeGen_C::visit ( const VectorReduce * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit_binop()

void Halide::Internal::CodeGen_C::visit_binop ( Type t,
const Expr & a,
const Expr & b,
const char * op )
protected

◆ visit_relop()

void Halide::Internal::CodeGen_C::visit_relop ( Type t,
const Expr & a,
const Expr & b,
const char * scalar_op,
const char * vector_op )
protected

◆ with_sep()

template<typename T >
static std::string Halide::Internal::CodeGen_C::with_sep ( const std::vector< T > & v,
const std::string & sep )
inlinestaticprotected

Definition at line 262 of file CodeGen_C.h.

Referenced by with_commas().

◆ with_commas()

template<typename T >
static std::string Halide::Internal::CodeGen_C::with_commas ( const std::vector< T > & v)
inlinestaticprotected

Definition at line 274 of file CodeGen_C.h.

References with_sep().

◆ is_stack_private_to_thread()

virtual bool Halide::Internal::CodeGen_C::is_stack_private_to_thread ( ) const
protectedvirtual

Some architectures have private memory for the call stack; this means a thread cannot hand pointers to stack memory to another thread.

Returning true here flag forces heap allocation of things that might be shared, such as closures and any buffer that may be used in a parallel context.

◆ emit_argv_wrapper()

void Halide::Internal::CodeGen_C::emit_argv_wrapper ( const std::string & function_name,
const std::vector< LoweredArgument > & args )
protected

◆ emit_metadata_getter()

void Halide::Internal::CodeGen_C::emit_metadata_getter ( const std::string & function_name,
const std::vector< LoweredArgument > & args,
const MetadataNameMap & metadata_name_map )
protected

◆ emit_constexpr_function_info()

void Halide::Internal::CodeGen_C::emit_constexpr_function_info ( const std::string & function_name,
const std::vector< LoweredArgument > & args,
const MetadataNameMap & metadata_name_map )
protected

◆ emit_halide_free_helper()

void Halide::Internal::CodeGen_C::emit_halide_free_helper ( const std::string & alloc_name,
const std::string & free_function )
protected

◆ visit() [47/48]

void Halide::Internal::IRPrinter::visit ( const Block * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

◆ visit() [48/48]

void Halide::Internal::IRPrinter::visit ( const HoistedStorage * )
overrideprotectedvirtual

Reimplemented from Halide::Internal::IRVisitor.

Reimplemented in Halide::Internal::CodeGen_GPU_C.

Member Data Documentation

◆ integer_suffix_style

IntegerSuffixStyle Halide::Internal::CodeGen_C::integer_suffix_style = IntegerSuffixStyle::PlainC
protected

How to emit 64-bit integer constants.

Definition at line 65 of file CodeGen_C.h.

◆ id

std::string Halide::Internal::CodeGen_C::id
protected

An ID for the most recently generated ssa variable.

Definition at line 86 of file CodeGen_C.h.

◆ target

Target Halide::Internal::CodeGen_C::target
protected

The target being generated for.

Definition at line 89 of file CodeGen_C.h.

Referenced by get_target().

◆ output_kind

OutputKind Halide::Internal::CodeGen_C::output_kind
protected

Controls whether this instance is generating declarations or definitions and whether the interface us extern "C" or C++.

Definition at line 93 of file CodeGen_C.h.

Referenced by is_c_plus_plus_interface(), is_extern_decl(), and is_header().

◆ cache

std::map<std::string, std::string> Halide::Internal::CodeGen_C::cache
protected

A cache of generated values in scope.

Definition at line 96 of file CodeGen_C.h.

◆ allocations

Scope<Allocation> Halide::Internal::CodeGen_C::allocations
protected

Track the types of allocations to avoid unnecessary casts.

Definition at line 186 of file CodeGen_C.h.

◆ heap_allocations

Scope Halide::Internal::CodeGen_C::heap_allocations
protected

Track which allocations actually went on the heap.

Definition at line 189 of file CodeGen_C.h.

◆ have_user_context

bool Halide::Internal::CodeGen_C::have_user_context
protected

True if there is a void * __user_context parameter in the arguments.

Definition at line 192 of file CodeGen_C.h.

◆ extern_c_open

bool Halide::Internal::CodeGen_C::extern_c_open = false
protected

Track current calling convention scope.

Definition at line 195 of file CodeGen_C.h.

◆ uses_gpu_for_loops

bool Halide::Internal::CodeGen_C::uses_gpu_for_loops
protected

True if at least one gpu-based for loop is used.

Definition at line 198 of file CodeGen_C.h.

◆ forward_declared

std::set<const halide_handle_cplusplus_type *> Halide::Internal::CodeGen_C::forward_declared
protected

Track which handle types have been forward-declared already.

Definition at line 201 of file CodeGen_C.h.

◆ inside_atomic_mutex_node

bool Halide::Internal::CodeGen_C::inside_atomic_mutex_node = false
protected

Are we inside an atomic node that uses mutex locks? This is used for detecting deadlocks from nested atomics.

Definition at line 280 of file CodeGen_C.h.

◆ emit_atomic_stores

bool Halide::Internal::CodeGen_C::emit_atomic_stores = false
protected

Emit atomic store instructions?

Definition at line 283 of file CodeGen_C.h.

◆ using_vector_typedefs

bool Halide::Internal::CodeGen_C::using_vector_typedefs = false
protected

true if add_vector_typedefs() has been called.

Definition at line 286 of file CodeGen_C.h.


The documentation for this class was generated from the following file: