This commit is contained in:
Awni Hannun
2025-08-20 14:05:35 -07:00
committed by Awni Hannun
parent 9807ba0267
commit 9a6f20e067
2 changed files with 157 additions and 67 deletions

View File

@@ -734,6 +734,8 @@ class TestQuantized(mlx_tests.MLXTestCase):
for L, K, D, E, I, transpose, mode in parameters:
if mode == "mxfp4":
group_size = 32
else:
group_size = 64
K, D = (K, D) if transpose else (D, K)
ishape = (L, I)
xshape = (L, 1, 1, K)