flash-attention/csrc/flash_attn
2023-12-23 20:59:26 -08:00
..
src Implement deterministic backward (thanks to Meituan) 2023-12-23 17:57:36 -08:00
flash_api.cpp Don't dispatch to local if window size >= seqlen_k 2023-12-23 20:59:26 -08:00