module.h File Reference

Runtime container of the functions generated by TVM, This is used to support dynamically link, load and save functions from different convention under unified API. More...

#include <dmlc/io.h>
#include <tvm/runtime/c_runtime_api.h>
#include <tvm/runtime/container/string.h>
#include <tvm/runtime/memory.h>
#include <tvm/runtime/object.h>
#include <memory>
#include <mutex>
#include <string>
#include <unordered_map>
#include <vector>
#include <tvm/runtime/packed_func.h>

Include dependency graph for module.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
class	tvm::runtime::Module
	Module container of TVM. More...
class	tvm::runtime::ModuleNode
	Base container of module. More...

Namespaces
	tvm
	runtime implementation for LibTorch/TorchScript.
	tvm::runtime
	tvm::runtime::symbol
	namespace for constant symbols

Enumerations
enum	tvm::runtime::ModulePropertyMask : int { tvm::runtime::kBinarySerializable = 0b001 , tvm::runtime::kRunnable = 0b010 , tvm::runtime::kDSOExportable = 0b100 }
	Property of runtime module We classify the property of runtime module into the following categories. More...

Functions
bool	tvm::runtime::RuntimeEnabled (const String &target)
	Check if runtime module is enabled for target. More...
std::ostream &	tvm::runtime::operator<< (std::ostream &out, const Module &module)

Variables
constexpr const char *	tvm::runtime::symbol::tvm_get_c_metadata = "get_c_metadata"
	A PackedFunc that retrieves exported metadata. More...
constexpr const char *	tvm::runtime::symbol::tvm_module_ctx = "__tvm_module_ctx"
	Global variable to store module context. More...
constexpr const char *	tvm::runtime::symbol::tvm_dev_mblob = "__tvm_dev_mblob"
	Global variable to store device module blob. More...
constexpr const char *	tvm::runtime::symbol::tvm_set_device = "__tvm_set_device"
	global function to set device More...
constexpr const char *	tvm::runtime::symbol::tvm_global_barrier_state = "__tvm_global_barrier_state"
	Auxiliary counter to global barrier. More...
constexpr const char *	tvm::runtime::symbol::tvm_prepare_global_barrier = "__tvm_prepare_global_barrier"
	Prepare the global barrier before kernels that uses global barrier. More...
constexpr const char *	tvm::runtime::symbol::tvm_module_main = "__tvm_main__"
	Placeholder for the module's entry function. More...
constexpr const char *	tvm::runtime::symbol::tvm_param_prefix = "__tvm_param__"
	Prefix for parameter symbols emitted into the main program. More...
constexpr const char *	tvm::runtime::symbol::tvm_lookup_linked_param = "_lookup_linked_param"
	A PackedFunc that looks up linked parameters by storage_id. More...
constexpr const char *	tvm::runtime::symbol::tvm_entrypoint_suffix = "run"
	Model entrypoint generated as an interface to the AOT function outside of TIR. More...

Detailed Description

Runtime container of the functions generated by TVM, This is used to support dynamically link, load and save functions from different convention under unified API.