Skip to content

Commit 0f235ee

Browse files
authored
[TRITON_KERNELS][MATMUL] fixed heuristics (#8813)
1 parent 1c15d61 commit 0f235ee

File tree

1 file changed

+2
-0
lines changed
  • python/triton_kernels/triton_kernels

1 file changed

+2
-0
lines changed

python/triton_kernels/triton_kernels/matmul.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -294,6 +294,8 @@ def matmul(a, b, bias,
294294
# Currently we don't support tma if y is column major; may revisit later if this becomes an issue.
295295
(c is None or c.stride(-1) == 1) and
296296
(c_acc_in is None or c_acc_is_c) and
297+
# for simulated MXFP, not supported
298+
(b_scale is None or target_info.has_native_mxfp()) and
297299
# if ragged dimension is K, w must be either padded or row major to ensure alignment
298300
(ragged_dimension != "K" or b.stride(-1) == 1 or b_ragged_metadata.slice_sizes_divisibility is not None)
299301
)

0 commit comments

Comments
 (0)