squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
9e92a1f2d2
flash-attention/csrc/flash_attn
History
Tri Dao 9e92a1f2d2 Attempt to use atomicCAS to replace atomicAdd(bfloat16)
2022-10-23 16:22:43 -07:00
..
cutlass@319a389f42 Add Cutlass as submodule 2022-06-02 09:54:16 -07:00
src Attempt to use atomicCAS to replace atomicAdd(bfloat16) 2022-10-23 16:22:43 -07:00
fmha_api.cpp Split bwd on the seqlen_q dimension 2022-10-23 11:35:15 -07:00