tvm
|
Generic attribute names that can be attached to any function. More...
Variables | |
constexpr const char * | kCallingConv = "calling_conv" |
Indicates the special calling convention. More... | |
constexpr const char * | kTarget = "target" |
Compilation target of the function. More... | |
constexpr const char * | kGlobalSymbol = "global_symbol" |
Global linker symbol of the function in generated code. More... | |
constexpr const char * | kModuleName = "mod_name" |
Name of the module. More... | |
constexpr const char * | kExecutor = "executor" |
Executor targeted by the module. More... | |
constexpr const char * | kRuntime = "runtime" |
Runtime target of the module. More... | |
constexpr const char * | kWorkspaceMemoryPools = "workspace_memory_pools" |
workspace memory pools of the module More... | |
constexpr const char * | kConstantMemoryPools = "constant_memory_pools" |
constant memory pools of the module More... | |
constexpr const char * | kConstants = "constants" |
constexpr const char * | kExternalMods = "external_mods" |
All the runtime::Modules accumulated during compilation by external codegen. These modules must be either directly linked or captured in the final compilation artifact. More... | |
constexpr const char * | kConstNameToConstant = "const_name_to_constant" |
All the named runtime::NDArrays accumulated during compilation by external codegen. Generally the associated runtime::Module will indicate it requires bindings for these names, and during module initialization these bindings will be recovered from a ConstLoaderModule. See also kConstantsArray above, which is the analog for PrimFuncs. More... | |
constexpr const char * | kIsExternalCodegen = "is_external_codegen" |
A TargetKind attribute of type Bool . If true, then the target kind name also corresponds to an external codegen 'compiler' name. That name may be used: More... | |
constexpr const char * | kRelayToTIR = "RelayToTIR" |
A TargetKind attribute of type FTVMRelayToTIR . If set, then the target kind name also corresponds to an external codegen 'compiler' name, and the bound value is a Pass to apply before the TVM lowering. More... | |
Generic attribute names that can be attached to any function.
constexpr const char* tvm::attr::kCallingConv = "calling_conv" |
constexpr const char* tvm::attr::kConstantMemoryPools = "constant_memory_pools" |
constexpr const char* tvm::attr::kConstants = "constants" |
constexpr const char* tvm::attr::kConstNameToConstant = "const_name_to_constant" |
All the named runtime::NDArrays accumulated during compilation by external codegen. Generally the associated runtime::Module will indicate it requires bindings for these names, and during module initialization these bindings will be recovered from a ConstLoaderModule. See also kConstantsArray above, which is the analog for PrimFuncs.
Type: Map<String, runtime::NDArray>
constexpr const char* tvm::attr::kExecutor = "executor" |
constexpr const char* tvm::attr::kExternalMods = "external_mods" |
All the runtime::Modules accumulated during compilation by external codegen. These modules must be either directly linked or captured in the final compilation artifact.
Type: Array<runtime::Module>
constexpr const char* tvm::attr::kGlobalSymbol = "global_symbol" |
Global linker symbol of the function in generated code.
This option forces the code generator to name the function with the given.
For example, we could set a global_symbol of a function early to make sure that we can always refer to it by the symbol name in the generated DLL.
We should not set the attribute for local functions, so that the compiler can freely rename them.
A unique global symbol will be automatically assigned to each function in the module before the target code generation phase.
Type: String
constexpr const char* tvm::attr::kIsExternalCodegen = "is_external_codegen" |
A TargetKind
attribute of type Bool
. If true, then the target kind name also corresponds to an external codegen 'compiler' name. That name may be used:
get_partition_table
.attr::kCompiler
attribute to indicate the function is to be compiled by the external codegen path.The CollagePartition
pass uses this attribute to guide it's search over candidate partitions using external codegen.
See also Target::IsExternalCodegenFor
constexpr const char* tvm::attr::kModuleName = "mod_name" |
constexpr const char* tvm::attr::kRelayToTIR = "RelayToTIR" |
A TargetKind
attribute of type FTVMRelayToTIR
. If set, then the target kind name also corresponds to an external codegen 'compiler' name, and the bound value is a Pass
to apply before the TVM lowering.
See also Target::IsExternalCodegenFor
constexpr const char* tvm::attr::kRuntime = "runtime" |
constexpr const char* tvm::attr::kTarget = "target" |
constexpr const char* tvm::attr::kWorkspaceMemoryPools = "workspace_memory_pools" |