memory.h File Reference

Memory operations for TIRx. More...

#include <tvm/ir/op.h>
#include <tvm/ir/prim/expr.h>

Include dependency graph for memory.h:

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

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.

Functions
TVM_TIRX_DLL const Op &	tvm::tirx::alloc_tensor_op ()
	Allocate a tensor with exactly one TensorType in ty_args.
TVM_TIRX_DLL const Op &	tvm::tirx::decl_tensor_op ()
	Declare a tensor view with exactly one TensorType in ty_args.
TVM_TIRX_DLL const Op &	tvm::tirx::reinterpret_op ()
	Reinterpret the bits of a value as an explicit primitive or pointer type.
TVM_TIRX_DLL Expr	tvm::tirx::reinterpret (Type target_ty, Expr value, Location loc=UnknownLoc())
	perform reinterpret cast value to type.
PrimExpr	tvm::tirx::reinterpret (PrimType t, PrimExpr value, Location loc=UnknownLoc())
Expr	tvm::tirx::ConstHandle (int64_t value, Location loc=UnknownLoc())
	Construct an opaque pointer from its integer payload.
TVM_TIRX_DLL const Op &	tvm::tirx::address_of_op ()
	Return the address of a tensor element or addressable variable.
TVM_TIRX_DLL const Op &	tvm::tirx::isnullptr_op ()
	Test whether a pointer is null.
TVM_TIRX_DLL const Op &	tvm::tirx::prefetch_op ()
	Prefetch a memory address.
TVM_TIRX_DLL const Op &	tvm::tirx::ptr_byte_offset_op ()
	Cast a handle to a typed pointer after adding a byte offset.
TVM_TIRX_DLL const Op &	tvm::tirx::static_handle_op ()
	Create a function local static handle that iniitalizes to nullptr.
TVM_TIRX_DLL const Op &	tvm::tirx::handle_add_byte_offset_op ()
	Add a byte offset to an opaque pointer.
TVM_TIRX_DLL const Op &	tvm::tirx::atomic_add_op ()
	atomic add instruction, corresponding e.g. to atomicAdd in CUDA.
TVM_TIRX_DLL const Op &	tvm::tirx::assume_aligned_op ()
	Assume a tensor's base address has the given constant byte alignment.
TVM_TIRX_DLL const Op &	tvm::tirx::undef_op ()
	Returns an initialized but arbitrary value.
TVM_TIRX_DLL const Op &	tvm::tirx::masked_load_op ()
	Masked tensor load.
TVM_TIRX_DLL const Op &	tvm::tirx::masked_store_op ()
	Masked tensor store.
TVM_TIRX_DLL const Op &	tvm::tirx::tensor_data_ptr_op ()
	Project the physical pointer associated with a TensorVar definition.

Detailed Description

Memory operations for TIRx.