A frame that represents the PrimFunc containing TIR statements. More...

#include <frame.h>

Inheritance diagram for tvm::script::ir_builder::tirx::PrimFuncFrameNode:

Collaboration diagram for tvm::script::ir_builder::tirx::PrimFuncFrameNode:

Public Member Functions
	TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("script.ir_builder.tirx.PrimFuncFrame", PrimFuncFrameNode, TIRFrameNode)

void	ExitWithScope () final
	The method called when exiting RAII scope.

Public Member Functions inherited from tvm::script::ir_builder::tirx::TIRFrameNode
	TVM_FFI_DECLARE_OBJECT_INFO ("script.ir_builder.tirx.TIRFrame", TIRFrameNode, IRBuilderFrameNode)

Public Member Functions inherited from tvm::script::ir_builder::IRBuilderFrameNode
	TVM_FFI_DECLARE_OBJECT_INFO ("script.ir_builder.IRBuilderFrame", IRBuilderFrameNode, ffi::Object)

virtual	~IRBuilderFrameNode ()=default
	Default destructor.

virtual void	EnterWithScope ()
	The method called when entering RAII scope.

void	AddCallback (ffi::TypedFunction< void()> callback)
	Add a callback method invoked when exiting the RAII scope.

Static Public Member Functions
static void	RegisterReflection ()

Static Public Member Functions inherited from tvm::script::ir_builder::tirx::TIRFrameNode
static void	RegisterReflection ()

Static Public Member Functions inherited from tvm::script::ir_builder::IRBuilderFrameNode
static void	RegisterReflection ()

Public Attributes
ffi::Optional< ffi::String >	name
	The name of the block.

ffi::Array< tvm::tirx::Var >	args
	Function parameters.

bool	is_private
	Whether the PrimFunc is annotated as private.

ffi::Optional< Type >	ret_type
	The return type of the function.

ffi::Map< tvm::tirx::Var, tvm::tirx::BufferVar >	buffer_map
	Maps some parameters to specific buffer data structures.

ffi::Map< ffi::String, Any >	attrs
	Additional attributes storing the meta-data.

ffi::Map< tvm::tirx::Var, tvm::tirx::IterVar >	env_threads
	The variable map bound to thread env.

ffi::Array< tvm::tirx::BufferVar >	root_alloc_buffers
	The buffer allocated in root block.

bool	s_tir
	Whether this PrimFunc uses s_tir semantics (root SBlock wrap, parser layout default = None). Default (false) = tirx semantics.

bool	persistent
	Whether it is a persistent kernel.

Public Attributes inherited from tvm::script::ir_builder::tirx::TIRFrameNode
ffi::Array< tvm::tirx::Stmt >	stmts
	The Stmt within in this frame.

Public Attributes inherited from tvm::script::ir_builder::IRBuilderFrameNode
std::vector< ffi::TypedFunction< void()> >	callbacks
	A list of callbacks used when exiting the frame.

Additional Inherited Members
Static Public Attributes inherited from tvm::script::ir_builder::IRBuilderFrameNode
static constexpr const bool	_type_mutable = true

Detailed Description

A frame that represents the PrimFunc containing TIR statements.

See also: PrimFuncFrame

Member Function Documentation

◆ ExitWithScope()

void tvm::script::ir_builder::tirx::PrimFuncFrameNode::ExitWithScope ( )

finalvirtual

The method called when exiting RAII scope.

See also: tvm::support::With

Reimplemented from tvm::script::ir_builder::IRBuilderFrameNode.

◆ RegisterReflection()

static void tvm::script::ir_builder::tirx::PrimFuncFrameNode::RegisterReflection ( )

inlinestatic

◆ TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

tvm::script::ir_builder::tirx::PrimFuncFrameNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL	(	"script.ir_builder.tirx.PrimFuncFrame"	,
		PrimFuncFrameNode	,
		TIRFrameNode
	)

Member Data Documentation

◆ args

ffi::Array<tvm::tirx::Var> tvm::script::ir_builder::tirx::PrimFuncFrameNode::args

Function parameters.

◆ attrs

ffi::Map<ffi::String, Any> tvm::script::ir_builder::tirx::PrimFuncFrameNode::attrs

Additional attributes storing the meta-data.

◆ buffer_map

ffi::Map<tvm::tirx::Var, tvm::tirx::BufferVar> tvm::script::ir_builder::tirx::PrimFuncFrameNode::buffer_map

Maps some parameters to specific buffer data structures.

◆ env_threads

ffi::Map<tvm::tirx::Var, tvm::tirx::IterVar> tvm::script::ir_builder::tirx::PrimFuncFrameNode::env_threads

The variable map bound to thread env.

◆ is_private

bool tvm::script::ir_builder::tirx::PrimFuncFrameNode::is_private

Whether the PrimFunc is annotated as private.

◆ name

ffi::Optional<ffi::String> tvm::script::ir_builder::tirx::PrimFuncFrameNode::name

The name of the block.

◆ persistent

bool tvm::script::ir_builder::tirx::PrimFuncFrameNode::persistent

Whether it is a persistent kernel.

◆ ret_type

ffi::Optional<Type> tvm::script::ir_builder::tirx::PrimFuncFrameNode::ret_type

The return type of the function.

◆ root_alloc_buffers

ffi::Array<tvm::tirx::BufferVar> tvm::script::ir_builder::tirx::PrimFuncFrameNode::root_alloc_buffers

The buffer allocated in root block.

◆ s_tir

bool tvm::script::ir_builder::tirx::PrimFuncFrameNode::s_tir

Whether this PrimFunc uses s_tir semantics (root SBlock wrap, parser layout default = None). Default (false) = tirx semantics.

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

include/tvm/tirx/script/builder/frame.h

Public Member Functions

Static Public Member Functions

Public Attributes

Additional Inherited Members