flash-attention/csrc
Tri Dao 7fc39832e2 Use block_size=128 for headdim=128 on SM80
Previously we were using block_size=256.
2022-10-21 13:19:54 -07:00
..
flash_attn Use block_size=128 for headdim=128 on SM80 2022-10-21 13:19:54 -07:00