Skip to content

Commit 1e5b330

Browse files
[mxfp] remove col-major check for mx weight
1 parent 4f7a8b8 commit 1e5b330

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

python/triton_kernels/triton_kernels/matmul_ogs.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -334,7 +334,6 @@ def matmul_ogs(x, w, bias,
334334
w_scale = precision_config.weight_scale
335335
w_has_mx = w_scale is not None
336336
is_hopper_fp8 = is_cuda() and not target_info.cuda_capability_geq(10, 0) and bitwidth(w.dtype) == 8
337-
if w_has_mx: assert w.stride(-2) == 1, "`w` must be column-major when it has data-type mxfp"
338337
if is_hopper_fp8: assert w.stride(-2) == 1, "`w` must be column-major when it has data-type FP8 on capability < 10"
339338
if not isinstance(w, Tensor):
340339
# TODO: remove this code path; using uint8 for mxfp4 weight will bite us when we want to support uint8 for real

0 commit comments

Comments
 (0)