[Bugfix] Marlin 2:4 temp fix for large M dim (>256) (#10464)

Signed-off-by: Lucas Wilkinson <lwilkinson@neuralmagic.com>
2024-11-19 22:40:33 -05:00
parent d5b68aba2f
commit d200972e7f
2 changed files with 13 additions and 4 deletions
--- a/tests/kernels/test_marlin_gemm.py
+++ b/tests/kernels/test_marlin_gemm.py
@@ -50,6 +50,8 @@ MNK_FACTORS = [
    (13, 17, 67),
    (26, 37, 13),
    (67, 13, 11),
+    (257, 13, 11),
+    (658, 13, 11),
 ]

 DTYPES = [torch.float16, torch.bfloat16]