cooperative_tensor.h File Reference

Metal-owned matrix operations. More...

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

Include dependency graph for cooperative_tensor.h:

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::backend
namespace	tvm::backend::metal

Functions
TVM_TIRX_DLL const Op &	tvm::backend::metal::cooperative_tensor_fill_op ()
	Fill a cooperative_tensor with a given value.
TVM_TIRX_DLL const Op &	tvm::backend::metal::cooperative_tensor_load_op ()
	Load data from device or threadgroup memory into a cooperative_tensor.
TVM_TIRX_DLL const Op &	tvm::backend::metal::cooperative_tensor_store_op ()
	Store data from a cooperative_tensor to device or threadgroup memory.
TVM_TIRX_DLL const Op &	tvm::backend::metal::cooperative_tensor_multiply_accumulate_op ()
	Multiply and accumulate two matrices using cooperative_tensor (MetalPerformancePrimitives matmul2d).

Detailed Description

Metal-owned matrix operations.