flash-attention/csrc/flash_attn
Tri Dao d380e87fb6 Don't use Smem_dp_sum in backward pass
To reduce smem usage for SM75
2022-06-04 16:01:36 -07:00
..
cutlass@319a389f42 Add Cutlass as submodule 2022-06-02 09:54:16 -07:00
src Don't use Smem_dp_sum in backward pass 2022-06-04 16:01:36 -07:00
fmha_api.cpp Support Turing mma instructions 2022-06-03 16:58:44 -07:00