Function tvm::backend::metal::cooperative_tensor_multiply_accumulate_op

Function tvm::backend::metal::cooperative_tensor_multiply_accumulate_op#

Function Documentation#

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

Multiply and accumulate two matrices using cooperative_tensor (MetalPerformancePrimitives matmul2d).

Arguments, in order:

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

  • args[1]: index_d, The D fragment index.

  • args[2]: a, The A operand.

  • args[3]: index_a, The A fragment index.

  • args[4]: b, The B operand.

  • args[5]: index_b, The B fragment index.

  • args[6]: c, The C operand.

  • args[7]: index_c, The C fragment index.

  • args[8]: M, The M dimension.

  • args[9]: N, The N dimension.

  • args[10]: K, The K dimension.

  • args[11]: transpose_a, Whether to transpose A.

  • args[12]: transpose_b, Whether to transpose B.