squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
9c0e9ee86d
flash-attention/csrc/flash_attn
History
Tri Dao 656daef4ea Use Cute's local_tile to get gQ, gK, gV
2024-04-07 20:10:19 -07:00
..
src Use Cute's local_tile to get gQ, gK, gV 2024-04-07 20:10:19 -07:00
flash_api.cpp Transpose out when swapping seqlen_q and num_groups 2024-04-07 20:10:19 -07:00