feat: enable head_dim=256 for attention kernels (#132)
#35
| Job | Run time |
|---|---|
| 18s | |
| 0s | |
| 18s |
head_dim=256 for attention kernels (#132)
#35
| Job | Run time |
|---|---|
| 18s | |
| 0s | |
| 18s |