tvm::tirx::tile::callback Namespace Reference

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

Variable Documentation

kDeviceInitStmt

constexpr const char* tvm::tirx::tile::callback::kDeviceInitStmt = "device_init_stmt"

constexpr

The initialization statement of the operator. which will be inserted at the beginning of the kernel.

kHostInitStmt

constexpr const char* tvm::tirx::tile::callback::kHostInitStmt = "host_init_stmt"

constexpr

The initialization statement of the operator. which will be inserted at the beginning of the kernel.

kPostTensorDefStmt

constexpr const char* tvm::tirx::tile::callback::kPostTensorDefStmt = "post_tensor_def_stmt"

constexpr

Statements to be inserted after a specific tensor's definition (DeclTensor/AllocTensor). Stored as Map<TensorVar, ffi::Array<Stmt>>.

kPrivateAlloc

constexpr const char* tvm::tirx::tile::callback::kPrivateAlloc = "private_alloc"

constexpr

The allocation Bind statements requested by the operator.