Skip to content

Commit d5de496

Browse files
authored
[KERNELS] Remove incorrect check (#7583)
This check is already done in the swizzling/unswizzling function so no need to repeat it here.
1 parent 7708b9c commit d5de496

File tree

1 file changed

+0
-4
lines changed

1 file changed

+0
-4
lines changed

python/triton_kernels/triton_kernels/matmul_ogs.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,6 @@
1616
from .matmul_ogs_details.opt_flags import make_opt_flags, update_opt_flags_constraints
1717
from .specialize import specialize
1818
from .tensor import Storage, Tensor, FP4, bitwidth, wrap_torch_tensor
19-
from .tensor_details import layout
2019

2120

2221
@dataclass(frozen=True)
@@ -403,9 +402,6 @@ def matmul_ogs(x, w, bias,
403402
assert K == x.shape[-1]
404403
if x.ndim == 3 and w.ndim == 3:
405404
assert x.shape[0] == w.shape[0]
406-
if isinstance(w.storage.layout, layout.HopperMXValueLayout):
407-
if not (K % 64 == 0 and N % 64 == 0):
408-
raise NotImplementedError("MXFP4 weight matrix dimensions must be divisible by 64 on hopper")
409405
# compute optimization flags
410406
out_dtype = precision_config.out_dtype or x.dtype
411407
can_use_tma = x.storage.is_tma_compliant() and \

0 commit comments

Comments
 (0)