flash-attention/csrc/flash_attn
2024-02-10 01:03:00 -08:00
..
src Preprocessor switches to control functionality (#788) 2024-01-29 20:44:23 -08:00
flash_api.cpp Fix dv = torch::empty_like(k) for mha_bwd_varlen as well 2024-02-10 01:03:00 -08:00