flash-attention/csrc/flash_attn
Kirthi Shankar Sivamani 31018c5fa0 Support CUDA graph capture
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
2023-04-12 16:53:22 -07: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 Support CUDA graph capture 2023-04-12 16:53:22 -07:00