squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
db2f80692c
flash-attention/csrc/flash_attn
History
Tri Dao db2f80692c Write zero to out / grad if seqlen_q or seqlen_k is zero
2023-11-19 22:20:01 -08:00
..
src Write zero to out / grad if seqlen_q or seqlen_k is zero 2023-11-19 22:20:01 -08:00
flash_api.cpp Write zero to out / grad if seqlen_q or seqlen_k is zero 2023-11-19 22:20:01 -08:00