squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
7d25a4ec4f
flash-attention/csrc/flash_attn
History
Kirthi Shankar Sivamani 7d25a4ec4f Handle FlashAttnQKVPackedSplitFunc by making rng_state optional in backward 
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
2023-04-13 06:25:52 +00:00
..
cutlass@319a389f42 Add Cutlass as submodule 2022-06-02 09:54:16 -07:00
src Support CUDA graph capture 2023-04-12 16:53:22 -07:00
fmha_api.cpp Handle FlashAttnQKVPackedSplitFunc by making rng_state optional in backward 2023-04-13 06:25:52 +00:00