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

#include <CodeGen_C.h>

Inheritance diagram for Halide::Internal::CodeGen_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 Target &	get_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_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	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< Allocation >	allocations
	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.

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

enum Halide::Internal::CodeGen_C::OutputKind

Enumerator
CHeader
CPlusPlusHeader
CImplementation
CPlusPlusImplementation
CExternDecl
CPlusPlusExternDecl
CPlusPlusFunctionInfoHeader

Definition at line 29 of file CodeGen_C.h.

◆ IntegerSuffixStyle

enum class Halide::Internal::CodeGen_C::IntegerSuffixStyle

strongprotected

Enumerator
PlainC
OpenCL
HLSL

Definition at line 58 of file CodeGen_C.h.

◆ AppendSpaceIfNeeded

enum Halide::Internal::CodeGen_C::AppendSpaceIfNeeded

protected

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:

src/CodeGen_C.h

Classes

Public Types

Public Member Functions

Static Public Member Functions

Protected Types

Protected Member Functions

Static Protected Member Functions

Protected Attributes

Detailed Description

Member Enumeration Documentation

◆ OutputKind

◆ IntegerSuffixStyle

◆ AppendSpaceIfNeeded

Constructor & Destructor Documentation

◆ CodeGen_C()

◆ ~CodeGen_C()

Member Function Documentation

◆ compile() [1/3]

◆ get_target()

◆ test()

◆ compile() [2/3]

◆ compile() [3/3]

◆ preprocess_function_body()

◆ print_expr()

◆ print_cast_expr()

◆ print_stmt()

◆ create_assertion() [1/2]

◆ create_assertion() [2/2]

◆ scalarize_vector_reduce()

◆ print_type()

◆ print_reinterpret()

◆ print_name()

◆ add_platform_prologue()

◆ add_vector_typedefs()

◆ print_extern_call()

◆ print_scalarized_expr()

◆ print_assignment()

◆ print_heap_free()

◆ is_header()

◆ is_extern_decl()

◆ is_header_or_extern_decl()

◆ is_c_plus_plus_interface()

◆ open_scope()

◆ close_scope()

◆ forward_declare_type_if_needed()

◆ set_name_mangling_mode()

◆ visit() [1/48]

◆ visit() [2/48]

◆ visit() [3/48]

◆ visit() [4/48]

◆ visit() [5/48]

◆ visit() [6/48]

◆ visit() [7/48]

◆ visit() [8/48]

◆ visit() [9/48]

◆ visit() [10/48]

◆ visit() [11/48]

◆ visit() [12/48]

◆ visit() [13/48]

◆ visit() [14/48]

◆ visit() [15/48]

◆ visit() [16/48]

◆ visit() [17/48]

◆ visit() [18/48]

◆ visit() [19/48]

◆ visit() [20/48]

◆ visit() [21/48]

◆ visit() [22/48]

◆ visit() [23/48]

◆ visit() [24/48]

◆ visit() [25/48]

◆ visit() [26/48]

◆ visit() [27/48]

◆ visit() [28/48]

◆ visit() [29/48]

◆ visit() [30/48]

◆ visit() [31/48]

◆ visit() [32/48]

◆ visit() [33/48]

◆ visit() [34/48]