squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
6f706eff96
flash-attention/csrc/flash_attn
History
Tri Dao 6f706eff96 Make Softmax an object
2024-01-19 16:09:31 -08:00
..
src Make Softmax an object 2024-01-19 16:09:31 -08:00
flash_api.cpp Don't dispatch to local if window size >= seqlen_k 2023-12-23 20:59:26 -08:00