squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
b28ec236df
flash-attention/csrc/flash_attn
History
Tri Dao 26d7d92f3d Fix splitKV combine function when local LSEs are all -inf
2023-09-03 11:39:09 -07:00
..
src Fix splitKV combine function when local LSEs are all -inf 2023-09-03 11:39:09 -07:00
flash_api.cpp Implement splitKV attention 2023-08-29 00:58:29 -07:00