squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
a4f148b6ab
flash-attention/csrc/flash_attn
History
Tri Dao a4f148b6ab Fix masking of bwd when seqlen is not divisible by 128
2023-07-31 17:46:34 -07:00
..
src Fix masking of bwd when seqlen is not divisible by 128 2023-07-31 17:46:34 -07:00
flash_api.cpp Enable CUDA graphs (#386) 2023-07-27 16:11:34 -07:00