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

Signed-off-by: Lucas Wilkinson <lwilkinson@neuralmagic.com>
This commit is contained in:
Lucas Wilkinson
2024-11-19 19:40:33 -08:00
committed by GitHub
parent d5b68aba2f
commit d200972e7f
2 changed files with 13 additions and 4 deletions
+2
View File
@@ -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]