tvm
Loading...
Searching...
No Matches
Classes | Namespaces | Variables
context.h File Reference
#include <tvm/target/target.h>
#include <tvm/tirx/stmt.h>
#include <tvm/tirx/tile/exec_scope.h>
Include dependency graph for context.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

class  tvm::tirx::tile::DispatchContextObj
 The context information of the kernel required by op dispatch. More...
 
class  tvm::tirx::tile::DispatchContext
 Managed reference to DispatchContextObj. More...
 

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.
 
namespace  tvm::tirx::tile
 Dispatch interface for lowering tile instruction Calls.
 
namespace  tvm::tirx::tile::callback
 

Variables

constexpr const char * tvm::tirx::tile::callback::kPrivateAlloc = "private_alloc"
 The allocation Bind statements requested by the operator.
 
constexpr const char * tvm::tirx::tile::callback::kDeviceInitStmt = "device_init_stmt"
 The initialization statement of the operator. which will be inserted at the beginning of the kernel.
 
constexpr const char * tvm::tirx::tile::callback::kHostInitStmt = "host_init_stmt"
 The initialization statement of the operator. which will be inserted at the beginning of the kernel.
 
constexpr const char * tvm::tirx::tile::callback::kPostTensorDefStmt = "post_tensor_def_stmt"
 Statements to be inserted after a specific tensor's definition (DeclTensor/AllocTensor). Stored as Map<TensorVar, ffi::Array<Stmt>>.