tvm_tirx.tirx.script.gpu_load_matrix_sync

tvm_tirx.tirx.script.gpu_load_matrix_sync#

tvm_tirx.tirx.script.gpu_load_matrix_sync(fragment, m, n, k, index, buffer_ptr, stride, layout, *, ty=None, loc=UNKNOWN_LOC)#

TVM intrinsic for tensor core load operators

Parameters:

  • fragment (Var) – The wmma fragment.

  • m (UIntImm) – The shape of wmma fragment.

  • n (UIntImm) – The shape of wmma fragment.

  • k (UIntImm) – The shape of wmma fragment.

  • index (Expr) – The fragment index.

  • buffer_ptr (Expr) – The fragment buffer pointer.

  • stride (Expr) – The fragment stride.

  • layout (Literal["row_major", "column_major"]) – The fragment layout.

  • loc (Location)

Returns:

call (Expr) – The call expression.