diff --git a/tests/unittest/_torch/thop/test_scaled_mm.py b/tests/unittest/_torch/thop/test_scaled_mm.py
index f3cb8ea856c0..32dce08a958f 100644
--- a/tests/unittest/_torch/thop/test_scaled_mm.py
+++ b/tests/unittest/_torch/thop/test_scaled_mm.py
@@ -31,7 +31,7 @@
 )
 @pytest.mark.parametrize(
     "m",
-    [2048, 12, 228],
+    [2048, 8, 228],
 )
 @pytest.mark.parametrize(
     "output_dtype",