Revert "Attempt different packing"

This reverts commit e4b587819c.
This commit is contained in:
Angelos Katharopoulos
2024-12-13 23:23:21 -08:00
parent e4b587819c
commit cb358dbdda
4 changed files with 24 additions and 32 deletions

View File

@@ -7,7 +7,7 @@ from time_utils import time_fn
D = 8192
group_size = 64
bits = 3
bits = 4
dtype = mx.float16
loops = 100