A reference-counted handle to Halide's internal representation of a function. More...

#include <Function.h>

Classes
struct	Compare
	This lets you use a Function as a key in a map of the form map<Function, Foo, Function::Compare> More...

Public Member Functions
	Function ()=default
	Construct a new function with no definitions and no name. More...

	Function (const std::string &n)
	Construct a new function with the given name. More...

	Function (const std::vector< Type > &required_types, int required_dims, const std::string &n)
	Construct a new function with the given name, with a requirement that it can only represent Expr(s) of the given type(s), and must have exactly the give nnumber of dimensions. More...

	Function (const FunctionPtr &)
	Construct a Function from an existing FunctionContents pointer. More...

FunctionPtr	get_contents () const
	Get a handle on the halide function contents that this Function represents. More...

void	deep_copy (const FunctionPtr &copy, std::map< FunctionPtr, FunctionPtr > &copied_map) const
	Deep copy this Function into 'copy'. More...

void	deep_copy (std::string name, const FunctionPtr &copy, std::map< FunctionPtr, FunctionPtr > &copied_map) const

void	define (const std::vector< std::string > &args, std::vector< Expr > values)
	Add a pure definition to this function. More...

void	define_update (const std::vector< Expr > &args, std::vector< Expr > values)
	Add an update definition to this function. More...

void	accept (IRVisitor *visitor) const
	Accept a visitor to visit all of the definitions and arguments of this function. More...

void	mutate (IRMutator *mutator)
	Accept a mutator to mutator all of the definitions and arguments of this function. More...

const std::string &	name () const
	Get the name of the function. More...

const std::string &	origin_name () const
	If this is a wrapper of another func, created by a chain of in or clone_in calls, returns the name of the original Func. More...

Definition &	definition ()
	Get a mutable handle to the init definition. More...

const Definition &	definition () const
	Get the init definition. More...

const std::vector< std::string > &	args () const
	Get the pure arguments. More...

int	dimensions () const
	Get the dimensionality. More...

int	outputs () const
	Get the number of outputs. More...

const std::vector< Type > &	output_types () const
	Get the types of the outputs. More...

const std::vector< Type > &	required_types () const
	Get the type constaints on the outputs (if any). More...

int	required_dimensions () const
	Get the dimensionality constaints on the outputs (if any). More...

const std::vector< Expr > &	values () const
	Get the right-hand-side of the pure definition. More...

bool	has_pure_definition () const
	Does this function have a pure definition? More...

bool	is_pure () const
	Does this function only have a pure definition? More...

bool	can_be_inlined () const
	Is it legal to inline this function? More...

FuncSchedule &	schedule ()
	Get a handle to the function-specific schedule for the purpose of modifying it. More...

const FuncSchedule &	schedule () const
	Get a const handle to the function-specific schedule for inspecting it. More...

const std::vector< Parameter > &	output_buffers () const
	Get a handle on the output buffer used for setting constraints on it. More...

StageSchedule &	update_schedule (int idx=0)
	Get a mutable handle to the stage-specfic schedule for the update stage. More...

Definition &	update (int idx=0)
	Get a mutable handle to this function's update definition at index 'idx'. More...

const Definition &	update (int idx=0) const
	Get a const reference to this function's update definition at index 'idx'. More...

const std::vector< Definition > &	updates () const
	Get a const reference to this function's update definitions. More...

bool	has_update_definition () const
	Does this function have an update definition? More...

bool	has_extern_definition () const
	Check if the function has an extern definition. More...

NameMangling	extern_definition_name_mangling () const
	Get the name mangling specified for the extern definition. More...

Expr	make_call_to_extern_definition (const std::vector< Expr > &args, const Target &t) const
	Make a call node to the extern definition. More...

Expr	extern_definition_proxy_expr () const
	Get the proxy Expr for the extern stage. More...

Expr &	extern_definition_proxy_expr ()

void	define_extern (const std::string &function_name, const std::vector< ExternFuncArgument > &args, const std::vector< Type > &types, const std::vector< Var > &dims, NameMangling mangling, DeviceAPI device_api)
	Add an external definition of this Func. More...

const std::vector< ExternFuncArgument > &	extern_arguments () const
	Retrive the arguments of the extern definition. More...

std::vector< ExternFuncArgument > &	extern_arguments ()

const std::string &	extern_function_name () const
	Get the name of the extern function called for an extern definition. More...

DeviceAPI	extern_function_device_api () const
	Get the DeviceAPI declared for an extern function. More...

bool	same_as (const Function &other) const
	Test for equality of identity. More...

const std::string &	debug_file () const
	Get a const handle to the debug filename. More...

std::string &	debug_file ()
	Get a handle to the debug filename. More...

	operator ExternFuncArgument () const
	Use an an extern argument to another function. More...

void	trace_loads ()
	Tracing calls and accessors, passed down from the Func equivalents. More...

void	trace_stores ()

void	trace_realizations ()

void	add_trace_tag (const std::string &trace_tag)

bool	is_tracing_loads () const

bool	is_tracing_stores () const

bool	is_tracing_realizations () const

const std::vector< std::string > &	get_trace_tags () const

void	lock_loop_levels ()
	Replace this Function's LoopLevels with locked copies that cannot be mutated further. More...

void	freeze ()
	Mark function as frozen, which means it cannot accept new definitions. More...

bool	frozen () const
	Check if a function has been frozen. More...

Function	new_function_in_same_group (const std::string &)
	Make a new Function with the same lifetime as this one, and return a strong reference to it. More...

void	add_wrapper (const std::string &f, Function &wrapper)
	Mark calls of this function by 'f' to be replaced with its wrapper during the lowering stage. More...

const std::map< std::string, FunctionPtr > &	wrappers () const

const Call *	is_wrapper () const
	Check if a Function is a trivial wrapper around another Function, Buffer, or Parameter. More...

Function &	substitute_calls (const std::map< FunctionPtr, FunctionPtr > &substitutions)
	Replace every call to Functions in 'substitutions' keys by all Exprs referenced in this Function to call to their substitute Functions (i.e. More...

Function &	substitute_calls (const Function &orig, const Function &substitute)

bool	is_pure_arg (const std::string &name) const
	Return true iff the name matches one of the Function's pure args. More...

void	check_types (const Expr &e) const
	If the Function has type requirements, check that the given argument is compatible with them. More...

void	check_types (const Tuple &t) const

void	check_types (const Type &t) const

void	check_types (const std::vector< Expr > &exprs) const

void	check_types (const std::vector< Type > &types) const

void	check_dims (int dims) const
	If the Function has dimension requirements, check that the given argument is compatible with them. More...

void	create_output_buffers (const std::vector< Type > &types, int dims) const
	Define the output buffers. More...

Detailed Description

A reference-counted handle to Halide's internal representation of a function.

Similar to a front-end Func object, but with no syntactic sugar to help with definitions.

Definition at line 39 of file Function.h.

Constructor & Destructor Documentation

◆ Function() [1/4]

Halide::Internal::Function::Function ( )

default

Construct a new function with no definitions and no name.

This constructor only exists so that you can make vectors of functions, etc.

◆ Function() [2/4]

Halide::Internal::Function::Function ( const std::string & n )

explicit

Construct a new function with the given name.

◆ Function() [3/4]

Halide::Internal::Function::Function	(	const std::vector< Type > &	required_types,
		int	required_dims,
		const std::string &	n
	)

explicit

Construct a new function with the given name, with a requirement that it can only represent Expr(s) of the given type(s), and must have exactly the give nnumber of dimensions.

required_types.empty() means there are no constraints on the type(s). required_dims == AnyDims means there are no constraints on the dimensions.

◆ Function() [4/4]

Halide::Internal::Function::Function ( const FunctionPtr & )

explicit

Construct a Function from an existing FunctionContents pointer.

Must be non-null

Member Function Documentation

◆ get_contents()

FunctionPtr Halide::Internal::Function::get_contents ( ) const

inline

Get a handle on the halide function contents that this Function represents.

Definition at line 73 of file Function.h.

◆ deep_copy() [1/2]

void Halide::Internal::Function::deep_copy	(	const FunctionPtr &	copy,
		std::map< FunctionPtr, FunctionPtr > &	copied_map
	)		const

Deep copy this Function into 'copy'.

It recursively deep copies all called functions, schedules, update definitions, extern func arguments, specializations, and reduction domains. This method does not deep-copy the Parameter objects. This method also takes a map of <old Function, deep-copied version> as input and would use the deep-copied Function from the map if exists instead of creating a new deep-copy to avoid creating deep-copies of the same Function multiple times. If 'name' is specified, copy's name will be set to that.

◆ deep_copy() [2/2]

void Halide::Internal::Function::deep_copy	(	std::string	name,
		const FunctionPtr &	copy,
		std::map< FunctionPtr, FunctionPtr > &	copied_map
	)		const

◆ define()

void Halide::Internal::Function::define	(	const std::vector< std::string > &	args,
		std::vector< Expr >	values
	)

Add a pure definition to this function.

It may not already have a definition. All the free variables in 'value' must appear in the args list. 'value' must not depend on any reduction domain

◆ define_update()

void Halide::Internal::Function::define_update	(	const std::vector< Expr > &	args,
		std::vector< Expr >	values
	)

Add an update definition to this function.

It must already have a pure definition but not an update definition, and the length of args must match the length of args used in the pure definition. 'value' must depend on some reduction domain, and may contain variables from that domain as well as pure variables. Any pure variables must also appear as Variables in the args array, and they must have the same name as the pure definition's argument in the same index.

◆ accept()

void Halide::Internal::Function::accept ( IRVisitor * visitor ) const

Accept a visitor to visit all of the definitions and arguments of this function.

◆ mutate()

void Halide::Internal::Function::mutate ( IRMutator * mutator )

Accept a mutator to mutator all of the definitions and arguments of this function.

◆ name()

const std::string& Halide::Internal::Function::name ( ) const

Get the name of the function.

Referenced by Halide::Internal::Autoscheduler::LoopNestParser::is_in_partial_schedule().

◆ origin_name()

const std::string& Halide::Internal::Function::origin_name ( ) const

If this is a wrapper of another func, created by a chain of in or clone_in calls, returns the name of the original Func.

Otherwise returns the name.

◆ definition() [1/2]

Definition& Halide::Internal::Function::definition ( )

Get a mutable handle to the init definition.

◆ definition() [2/2]

const Definition& Halide::Internal::Function::definition ( ) const

Get the init definition.

◆ args()

const std::vector<std::string>& Halide::Internal::Function::args ( ) const

Get the pure arguments.

◆ dimensions()

int Halide::Internal::Function::dimensions ( ) const

Get the dimensionality.

◆ outputs()

int Halide::Internal::Function::outputs ( ) const

Get the number of outputs.

◆ output_types()

const std::vector<Type>& Halide::Internal::Function::output_types ( ) const

Get the types of the outputs.

◆ required_types()

const std::vector<Type>& Halide::Internal::Function::required_types ( ) const

Get the type constaints on the outputs (if any).

◆ required_dimensions()

int Halide::Internal::Function::required_dimensions ( ) const

Get the dimensionality constaints on the outputs (if any).

◆ values()

const std::vector<Expr>& Halide::Internal::Function::values ( ) const

Get the right-hand-side of the pure definition.

Returns an empty vector if there is no pure definition.

◆ has_pure_definition()

bool Halide::Internal::Function::has_pure_definition ( ) const

Does this function have a pure definition?

Referenced by is_pure().

◆ is_pure()

bool Halide::Internal::Function::is_pure ( ) const

inline

Does this function only have a pure definition?

Definition at line 155 of file Function.h.

References has_extern_definition(), has_pure_definition(), and has_update_definition().

◆ can_be_inlined()

bool Halide::Internal::Function::can_be_inlined ( ) const

Is it legal to inline this function?

◆ schedule() [1/2]

FuncSchedule& Halide::Internal::Function::schedule ( )

Get a handle to the function-specific schedule for the purpose of modifying it.

◆ schedule() [2/2]

const FuncSchedule& Halide::Internal::Function::schedule ( ) const

Get a const handle to the function-specific schedule for inspecting it.

◆ output_buffers()

const std::vector<Parameter>& Halide::Internal::Function::output_buffers ( ) const

Get a handle on the output buffer used for setting constraints on it.

◆ update_schedule()

StageSchedule& Halide::Internal::Function::update_schedule ( int idx = 0 )

Get a mutable handle to the stage-specfic schedule for the update stage.

◆ update() [1/2]

Definition& Halide::Internal::Function::update ( int idx = 0 )

Get a mutable handle to this function's update definition at index 'idx'.

◆ update() [2/2]

const Definition& Halide::Internal::Function::update ( int idx = 0 ) const

Get a const reference to this function's update definition at index 'idx'.

◆ updates()

const std::vector<Definition>& Halide::Internal::Function::updates ( ) const

Get a const reference to this function's update definitions.

◆ has_update_definition()

bool Halide::Internal::Function::has_update_definition ( ) const

Does this function have an update definition?

Referenced by Halide::SimdOpCheckTest::check_one(), and is_pure().

◆ has_extern_definition()

bool Halide::Internal::Function::has_extern_definition ( ) const

Check if the function has an extern definition.

Referenced by is_pure().

◆ extern_definition_name_mangling()

NameMangling Halide::Internal::Function::extern_definition_name_mangling ( ) const

Get the name mangling specified for the extern definition.

◆ make_call_to_extern_definition()

Expr Halide::Internal::Function::make_call_to_extern_definition	(	const std::vector< Expr > &	args,
		const Target &	t
	)		const

Make a call node to the extern definition.

An error if the function has no extern definition.

◆ extern_definition_proxy_expr() [1/2]

Expr Halide::Internal::Function::extern_definition_proxy_expr ( ) const

Get the proxy Expr for the extern stage.

This is an expression known to have the same data access pattern as the extern stage. It must touch at least all of the memory that the extern stage does, though it is permissible for it to be conservative and touch a superset. For most Functions, including those with extern definitions, this will be an undefined Expr.

◆ extern_definition_proxy_expr() [2/2]

Expr& Halide::Internal::Function::extern_definition_proxy_expr ( )

◆ define_extern()

void Halide::Internal::Function::define_extern	(	const std::string &	function_name,
		const std::vector< ExternFuncArgument > &	args,
		const std::vector< Type > &	types,
		const std::vector< Var > &	dims,
		NameMangling	mangling,
		DeviceAPI	device_api
	)

Add an external definition of this Func.

◆ extern_arguments() [1/2]

const std::vector<ExternFuncArgument>& Halide::Internal::Function::extern_arguments ( ) const

Retrive the arguments of the extern definition.

◆ extern_arguments() [2/2]

std::vector<ExternFuncArgument>& Halide::Internal::Function::extern_arguments ( )

◆ extern_function_name()

const std::string& Halide::Internal::Function::extern_function_name ( ) const

Get the name of the extern function called for an extern definition.

◆ extern_function_device_api()

DeviceAPI Halide::Internal::Function::extern_function_device_api ( ) const

Get the DeviceAPI declared for an extern function.

◆ same_as()

bool Halide::Internal::Function::same_as ( const Function & other ) const

inline

Test for equality of identity.

Definition at line 236 of file Function.h.

References Halide::Internal::FunctionPtr::same_as().

◆ debug_file() [1/2]

const std::string& Halide::Internal::Function::debug_file ( ) const

Get a const handle to the debug filename.

◆ debug_file() [2/2]

std::string& Halide::Internal::Function::debug_file ( )

Get a handle to the debug filename.

◆ operator ExternFuncArgument()

Halide::Internal::Function::operator ExternFuncArgument ( ) const

Use an an extern argument to another function.

◆ trace_loads()

void Halide::Internal::Function::trace_loads ( )

Tracing calls and accessors, passed down from the Func equivalents.

◆ trace_stores()

void Halide::Internal::Function::trace_stores ( )

◆ trace_realizations()

void Halide::Internal::Function::trace_realizations ( )

◆ add_trace_tag()

void Halide::Internal::Function::add_trace_tag ( const std::string & trace_tag )

◆ is_tracing_loads()

bool Halide::Internal::Function::is_tracing_loads ( ) const

◆ is_tracing_stores()

bool Halide::Internal::Function::is_tracing_stores ( ) const

◆ is_tracing_realizations()

bool Halide::Internal::Function::is_tracing_realizations ( ) const

◆ get_trace_tags()

const std::vector<std::string>& Halide::Internal::Function::get_trace_tags ( ) const

◆ lock_loop_levels()

void Halide::Internal::Function::lock_loop_levels ( )

Replace this Function's LoopLevels with locked copies that cannot be mutated further.

◆ freeze()

void Halide::Internal::Function::freeze ( )

Mark function as frozen, which means it cannot accept new definitions.

◆ frozen()

bool Halide::Internal::Function::frozen ( ) const

Check if a function has been frozen.

If so, it is an error to add new definitions.

◆ new_function_in_same_group()

Function Halide::Internal::Function::new_function_in_same_group ( const std::string & )

Make a new Function with the same lifetime as this one, and return a strong reference to it.

Useful to create Functions which have circular references to this one - e.g. the wrappers produced by Func::in.

◆ add_wrapper()

void Halide::Internal::Function::add_wrapper	(	const std::string &	f,
		Function &	wrapper
	)

Mark calls of this function by 'f' to be replaced with its wrapper during the lowering stage.

If the string 'f' is empty, it means replace all calls to this function by all other functions (excluding itself) in the pipeline with the wrapper. This will also freeze 'wrapper' to prevent user from updating the values of the Function it wraps via the wrapper. See Func::in for more details.

◆ wrappers()

const std::map<std::string, FunctionPtr>& Halide::Internal::Function::wrappers ( ) const

◆ is_wrapper()

const Call* Halide::Internal::Function::is_wrapper ( ) const

Check if a Function is a trivial wrapper around another Function, Buffer, or Parameter.

Returns the Call node if it is. Otherwise returns null.

◆ substitute_calls() [1/2]

Function& Halide::Internal::Function::substitute_calls ( const std::map< FunctionPtr, FunctionPtr > & substitutions )

Replace every call to Functions in 'substitutions' keys by all Exprs referenced in this Function to call to their substitute Functions (i.e.

the corresponding values in 'substitutions' map).

◆ substitute_calls() [2/2]

Function& Halide::Internal::Function::substitute_calls	(	const Function &	orig,
		const Function &	substitute
	)

◆ is_pure_arg()

bool Halide::Internal::Function::is_pure_arg ( const std::string & name ) const

Return true iff the name matches one of the Function's pure args.

◆ check_types() [1/5]

void Halide::Internal::Function::check_types ( const Expr & e ) const

If the Function has type requirements, check that the given argument is compatible with them.

If not, assert-fail. (If there are no type requirements, do nothing.)

◆ check_types() [2/5]

void Halide::Internal::Function::check_types ( const Tuple & t ) const

◆ check_types() [3/5]

void Halide::Internal::Function::check_types ( const Type & t ) const

◆ check_types() [4/5]

void Halide::Internal::Function::check_types ( const std::vector< Expr > & exprs ) const

◆ check_types() [5/5]

void Halide::Internal::Function::check_types ( const std::vector< Type > & types ) const

◆ check_dims()

void Halide::Internal::Function::check_dims ( int dims ) const

If the Function has dimension requirements, check that the given argument is compatible with them.

If not, assert-fail. (If there are no dimension requirements, do nothing.)

◆ create_output_buffers()

void Halide::Internal::Function::create_output_buffers	(	const std::vector< Type > &	types,
		int	dims
	)		const

Define the output buffers.

If the Function has types specified, this can be called at any time. If not, it can only be called for a Function with a pure definition.

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

src/Function.h

Classes

Public Member Functions

Detailed Description

Constructor & Destructor Documentation

◆ Function() [1/4]

◆ Function() [2/4]

◆ Function() [3/4]

◆ Function() [4/4]

Member Function Documentation

◆ get_contents()

◆ deep_copy() [1/2]

◆ deep_copy() [2/2]

◆ define()

◆ define_update()

◆ accept()

◆ mutate()

◆ name()

◆ origin_name()

◆ definition() [1/2]

◆ definition() [2/2]

◆ args()

◆ dimensions()

◆ outputs()

◆ output_types()

◆ required_types()

◆ required_dimensions()

◆ values()

◆ has_pure_definition()

◆ is_pure()

◆ can_be_inlined()

◆ schedule() [1/2]

◆ schedule() [2/2]

◆ output_buffers()

◆ update_schedule()

◆ update() [1/2]

◆ update() [2/2]

◆ updates()

◆ has_update_definition()

◆ has_extern_definition()

◆ extern_definition_name_mangling()

◆ make_call_to_extern_definition()

◆ extern_definition_proxy_expr() [1/2]

◆ extern_definition_proxy_expr() [2/2]

◆ define_extern()

◆ extern_arguments() [1/2]

◆ extern_arguments() [2/2]

◆ extern_function_name()

◆ extern_function_device_api()

◆ same_as()

◆ debug_file() [1/2]

◆ debug_file() [2/2]

◆ operator ExternFuncArgument()

◆ trace_loads()

◆ trace_stores()

◆ trace_realizations()

◆ add_trace_tag()

◆ is_tracing_loads()

◆ is_tracing_stores()

◆ is_tracing_realizations()

◆ get_trace_tags()

◆ lock_loop_levels()

◆ freeze()

◆ frozen()

◆ new_function_in_same_group()

◆ add_wrapper()

◆ wrappers()

◆ is_wrapper()

◆ substitute_calls() [1/2]

◆ substitute_calls() [2/2]

◆ is_pure_arg()

◆ check_types() [1/5]

◆ check_types() [2/5]

◆ check_types() [3/5]

◆ check_types() [4/5]

◆ check_types() [5/5]

◆ check_dims()

◆ create_output_buffers()