#include <frame.h>

Inheritance diagram for tvm::script::ir_builder::tir::LaunchThreadFrameNode:

Collaboration diagram for tvm::script::ir_builder::tir::LaunchThreadFrameNode:

Public Member Functions
	TVM_DECLARE_FINAL_OBJECT_INFO (LaunchThreadFrameNode, TIRFrameNode)

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

Public Member Functions inherited from tvm::script::ir_builder::tir::TIRFrameNode
	TVM_DECLARE_BASE_OBJECT_INFO (TIRFrameNode, IRBuilderFrameNode)

Public Member Functions inherited from tvm::script::ir_builder::IRBuilderFrameNode
	TVM_DECLARE_BASE_OBJECT_INFO (IRBuilderFrameNode, runtime::Object)

virtual	~IRBuilderFrameNode ()=default
	Default destructor. More...

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

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

Static Public Member Functions
static void	RegisterReflection ()

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

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

Public Attributes
PrimExpr	extent
	The extent of environment thread. More...

String	attr_key
	The attribute key, could be either virtual_thread or thread_extent. More...

tvm::tir::IterVar	iter_var
	The iteration variable. More...

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

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

Static Public Attributes
static constexpr const char *	_type_key = "script.ir_builder.tir.LaunchThreadFrame"

Static Public Attributes inherited from tvm::script::ir_builder::tir::TIRFrameNode
static constexpr const char *	_type_key = "script.ir_builder.tir.TIRFrame"

Static Public Attributes inherited from tvm::script::ir_builder::IRBuilderFrameNode
static constexpr const char *	_type_key = "script.ir_builder.IRBuilderFrame"

Detailed Description

The LaunchThreadFrameNode.

Note: It is used only inside a PrimFunc.

Member Function Documentation

◆ ExitWithScope()

void tvm::script::ir_builder::tir::LaunchThreadFrameNode::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::tir::LaunchThreadFrameNode::RegisterReflection ( )

inlinestatic

◆ TVM_DECLARE_FINAL_OBJECT_INFO()

tvm::script::ir_builder::tir::LaunchThreadFrameNode::TVM_DECLARE_FINAL_OBJECT_INFO	(	LaunchThreadFrameNode	,
		TIRFrameNode
	)

Member Data Documentation

◆ _type_key

constexpr const char* tvm::script::ir_builder::tir::LaunchThreadFrameNode::_type_key = "script.ir_builder.tir.LaunchThreadFrame"

staticconstexpr

◆ attr_key

String tvm::script::ir_builder::tir::LaunchThreadFrameNode::attr_key

The attribute key, could be either virtual_thread or thread_extent.

◆ extent

PrimExpr tvm::script::ir_builder::tir::LaunchThreadFrameNode::extent

The extent of environment thread.

◆ iter_var

tvm::tir::IterVar tvm::script::ir_builder::tir::LaunchThreadFrameNode::iter_var

The iteration variable.

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

include/tvm/script/ir_builder/tir/frame.h

Public Member Functions

Static Public Member Functions

Public Attributes

Static Public Attributes