tvm
Public Member Functions | Static Public Member Functions | Public Attributes | List of all members
tvm::script::ir_builder::tir::LaunchThreadFrameNode Class Reference

The LaunchThreadFrameNode. More...

#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_FFI_DECLARE_OBJECT_INFO_FINAL ("script.ir_builder.tir.LaunchThreadFrame", 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_FFI_DECLARE_OBJECT_INFO ("script.ir_builder.tir.TIRFrame", TIRFrameNode, IRBuilderFrameNode)
 
- Public Member Functions inherited from tvm::script::ir_builder::IRBuilderFrameNode
 TVM_FFI_DECLARE_OBJECT_INFO ("script.ir_builder.IRBuilderFrame", 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...
 
ffi::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
ffi::Array< tvm::tir::Stmtstmts
 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...
 

Additional Inherited Members

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

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_FFI_DECLARE_OBJECT_INFO_FINAL()

tvm::script::ir_builder::tir::LaunchThreadFrameNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL ( "script.ir_builder.tir.LaunchThreadFrame"  ,
LaunchThreadFrameNode  ,
TIRFrameNode   
)

Member Data Documentation

◆ attr_key

ffi::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: