tvm_tirx.tirx.gpu_fill_fragment

tvm_tirx.tirx.gpu_fill_fragment(fragment, m, n, k, index, value, *, ty=None, loc=UNKNOWN_LOC)

TVM intrinsic for tensor core fill_fragment 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.

• value (Expr) – The value to be filled in fragment.

• loc (Location)

Returns:

call (Expr) – The call expression.