feat: enable head_dim=256
for attention kernels#132
Merged
yzh119 merged 8 commits intomainfrom head-dim-256Feb 25, 2024
+415-297
Commits
Commits on Feb 22, 2024
- committed
- committed
- committed
- committed
- committed