squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
c65b5106ac
flash-attention/csrc/flash_attn
History
Tri Dao c65b5106ac Fix Bwd NaN for varlen when seqlen_q >> seqlen_k and causal
2023-08-16 15:12:36 -07:00
..
src Fix Bwd NaN for varlen when seqlen_q >> seqlen_k and causal 2023-08-16 15:12:36 -07:00
flash_api.cpp Enable CUDA graphs (#386) 2023-07-27 16:11:34 -07:00