tvm_tirx.backend.metal.script.MetalNamespace

class tvm_tirx.backend.metal.script.MetalNamespace

Bases: object

The Metal intrinsics submodule.

static cooperative_tensor_fill(d, index, value, rows, cols, *, ty=None, loc=UNKNOWN_LOC)

Parameters:

• d (Var)

• index (Expr)

• value (Expr)

• rows (int)

• cols (int)

• loc (Location)

static cooperative_tensor_load(d, index, ptr, stride, rows, cols, transpose_matrix=False, mma_M=0, mma_N=0, mma_K=0, operand_role=0, *, ty=None, loc=UNKNOWN_LOC)

Parameters:

• d (Var)

• index (Expr)

• ptr (Expr)

• stride (Expr)

• rows (int)

• cols (int)

• transpose_matrix (bool)

• mma_M (int)

• mma_N (int)

• mma_K (int)

• operand_role (int)

• loc (Location)

static cooperative_tensor_store(d, index, ptr, stride, rows, cols, transpose_matrix=False, mma_M=0, mma_N=0, mma_K=0, operand_role=0, *, ty=None, loc=UNKNOWN_LOC)

Parameters:

• d (Expr)

• index (Expr)

• ptr (Expr)

• stride (Expr)

• rows (int)

• cols (int)

• transpose_matrix (bool)

• mma_M (int)

• mma_N (int)

• mma_K (int)

• operand_role (int)

• loc (Location)

static cooperative_tensor_multiply_accumulate(d, index_d, a, index_a, b, index_b, c, index_c, M, N, K, transpose_a=False, transpose_b=False, *, ty=None, loc=UNKNOWN_LOC)

Parameters:

• d (Var)

• index_d (Expr)

• a (Var)

• index_a (Expr)

• b (Var)

• index_b (Expr)

• c (Var)

• index_c (Expr)

• M (int)

• N (int)

• K (int)

• transpose_a (bool)

• transpose_b (bool)

• loc (Location)

static simd_shuffle(var, lane, *, ty=None, loc=UNKNOWN_LOC)

Shuffle a value from the selected SIMD lane.

Parameters:

loc (Location)

static simd_shuffle_up(var, delta, *, ty=None, loc=UNKNOWN_LOC)

Shuffle a value from a lower SIMD lane.

Parameters:

loc (Location)

static simd_shuffle_down(var, delta, *, ty=None, loc=UNKNOWN_LOC)

Shuffle a value from a higher SIMD lane.

Parameters:

loc (Location)