tvm::tirx::callback Namespace Reference

Variables
constexpr const char *	kPrivateAlloc = "private_alloc"
	The buffers allocated 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 *	kPostBufferDefStmt = "post_buffer_def_stmt"
	Statements to be inserted after a specific buffer's definition (DeclBuffer/AllocBuffer). Stored as Map<BufferVar, ffi::Array<Stmt>>.

Variable Documentation

kDeviceInitStmt

constexpr const char* tvm::tirx::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::callback::kHostInitStmt = "host_init_stmt"

constexpr

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

kPostBufferDefStmt

constexpr const char* tvm::tirx::callback::kPostBufferDefStmt = "post_buffer_def_stmt"

constexpr

Statements to be inserted after a specific buffer's definition (DeclBuffer/AllocBuffer). Stored as Map<BufferVar, ffi::Array<Stmt>>.

kPrivateAlloc

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

constexpr

The buffers allocated by the operator.