|
tvm
|
Analysis utilities and passes for TIR. More...
#include <tvm/ir/module.h>#include <tvm/ir/prim/expr.h>#include <tvm/ir/transform.h>#include <tvm/target/target.h>#include <tvm/tirx/function.h>#include <tvm/tirx/op_attr_types.h>#include <tvm/tirx/stmt.h>#include <string>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. | |
| namespace | tvm::tirx::transform |
Functions | |
| TVM_TIRX_DLL bool | tvm::tirx::VerifyMemory (const Function &func) |
| Verify if memory accesses are legal for a specific target device type. | |
| TVM_TIRX_DLL size_t | tvm::tirx::CalculateExprComplexity (const PrimExpr &expr) |
| Calculate the expression complexity based on number of symbols it contains. | |
| TVM_TIRX_DLL size_t | tvm::tirx::CalculateConstantBytes (const Function &func, int64_t constant_byte_alignment) |
| Calculate the constants size in bytes needed by the TIR allocates inside the TIR Function. | |
| TVM_TIRX_DLL size_t | tvm::tirx::CalculateWorkspaceBytes (const Function &func, int64_t workspace_byte_alignment) |
| Calculate the workspace size in bytes needed by the TIR allocates inside the TIR Function. | |
| TVM_TIRX_DLL bool | tvm::tirx::VerifyWellFormed (const Function &func, bool assert_mode=true) |
| Verify if the given TIR is well-formed. The verification includes: | |
| TVM_TIRX_DLL bool | tvm::tirx::VerifyWellFormed (const IRModule &mod, bool assert_mode=true) |
| Verify if the TIR in the given IRMOdule is well-formed. | |
| const FunctionNode * | tvm::tirx::FindEntryFunc (const IRModule &mod, GlobalVar *result_g_var) |
Find the entry function of the given IRModule, i.e, functions marked by tirx::attr::kIsEntryFunc, whose name is main or being the only Function. | |
| TVM_TIRX_DLL Pass | tvm::tirx::transform::VerifyMemory () |
| Pass variant of VerifyMemory. | |
Analysis utilities and passes for TIR.