tvm_tirx.tirx.script.gpu_mma_sync

Contents

tvm_tirx.tirx.script.gpu_mma_sync#

tvm_tirx.tirx.script.gpu_mma_sync(fragment_d, index_d, fragment_a, index_a, fragment_b, index_b, fragment_c, index_c, *, ty=None, loc=UNKNOWN_LOC)#

TVM intrinsic for tensor core mma_sync operators

Parameters:

  • fragment_d (Var) – The wmma fragment_d.

  • index_d (Expr) – The fragment_d index.

  • fragment_a (Var) – The wmma fragment_a.

  • index_a (Expr) – The fragment_a index.

  • fragment_b (Var) – The wmma fragment_b.

  • index_b (Expr) – The fragment_b index.

  • fragment_c (Var) – The wmma fragment_c.

  • index_c (Expr) – The fragment_c index.

  • loc (Location)

Returns:

call (Expr) – The call expression.

Contents