From fa89f0b150cb85588c160ada1bd42b3803d7c087 Mon Sep 17 00:00:00 2001 From: Awni Hannun Date: Tue, 5 Aug 2025 06:27:40 -0700 Subject: [PATCH] faster gather qmm sorted test (#2463) --- python/tests/test_quantized.py | 22 +++++++++++----------- 1 file changed, 11 insertions(+), 11 deletions(-) diff --git a/python/tests/test_quantized.py b/python/tests/test_quantized.py index de43ec26d5..90a57221fc 100644 --- a/python/tests/test_quantized.py +++ b/python/tests/test_quantized.py @@ -525,17 +525,17 @@ class TestQuantized(mlx_tests.MLXTestCase): parameters = [ # L, K, D, E, I, transpose - (128, 1024, 1024, 32, 4, True), - (128, 1024, 544, 32, 4, True), - (433, 1024, 1024, 32, 4, True), - (433, 1024, 555, 32, 4, True), - (433, 2048, 1024, 32, 4, True), - (128, 1024, 1024, 32, 4, False), - (128, 1024, 544, 32, 4, False), - (433, 1024, 1024, 32, 4, False), - (433, 1024, 544, 32, 4, False), - (433, 1024, 555, 32, 4, False), - (433, 2048, 1024, 32, 4, False), + (32, 512, 512, 4, 2, True), + (32, 512, 544, 4, 2, True), + (133, 512, 512, 4, 2, True), + (133, 512, 555, 4, 2, True), + (133, 512, 512, 4, 2, True), + (64, 512, 512, 4, 2, False), + (64, 512, 544, 4, 2, False), + (133, 512, 512, 4, 2, False), + (133, 512, 544, 4, 2, False), + (133, 512, 555, 4, 2, False), + (64, 512, 512, 4, 2, False), ] for L, K, D, E, I, transpose in parameters: K, D = (K, D) if transpose else (D, K)