|
tvm
|
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>>. | |