Function tvm::backend::metal::cooperative_tensor_store_op

Function tvm::backend::metal::cooperative_tensor_store_op#

Function Documentation#

const Op &tvm::backend::metal::cooperative_tensor_store_op()#

Store data from a cooperative_tensor to device or threadgroup memory.

Arguments, in order:

  • args[0]: d, The D operand.

  • args[1]: index, The index.

  • args[2]: ptr, The pointer.

  • args[3]: stride, The stride.

  • args[4]: rows, The number of rows.

  • args[5]: cols, The number of columns.

  • args[6]: transpose_matrix, Whether to transpose the matrix.

  • args[7]: mma_M, The M dimension of the matrix operation.

  • args[8]: mma_N, The N dimension of the matrix operation.

  • args[9]: mma_K, The K dimension of the matrix operation.

  • args[10]: operand_role, 0 for left (A), 1 for right (B), 2 for destination (C).