tvm
Loading...
Searching...
No Matches
Namespaces | Functions
region.h File Reference

Region operations for TIRx. More...

#include <tvm/ir/op.h>
#include <tvm/ir/prim/expr.h>
Include dependency graph for region.h:

Go to the source code of this file.

Namespaces

namespace  tvm
 An object that builds and maintains block scope and StmtSref mapping for Dependence analysis.
 
namespace  tvm::tirx
 Backend-defined execution hierarchy nodes shared by layouts and tile calls.
 

Functions

TVM_TIRX_DLL const Op & tvm::tirx::launch_thread_op ()
 Thread launch region: operands are a nonempty StringImm tag and a scalar signed or unsigned integer extent wider than one bit. The sole body parameter is a fresh thread-index PrimVar matching the extent type. Tags starting with vthread denote virtual threads. There are no attrs or results.
 
TVM_TIRX_DLL const Op & tvm::tirx::device_entry_op ()
 Mark a user-facing device entry containing device scope definitions. Takes no operands, body parameters, attributes or results.
 
TVM_TIRX_DLL const Op & tvm::tirx::device_context_op ()
 Supply lexical device context for allocation and packed-call lowering. Operands are integer device type and device ID; there are no body parameters, attributes or results. The region does not change the active runtime device.
 
TVM_TIRX_DLL const Op & tvm::tirx::compute_scope_op ()
 Outline the body as a CPU compute helper named by a StringImm operand. There are no body parameters, attributes or results.
 
TVM_TIRX_DLL const Op & tvm::tirx::parallel_launch_op ()
 Launch a CPU worker team around parallel loops and team barriers. Takes no operands, body parameters, attributes or results.
 

Detailed Description

Region operations for TIRx.