flash-attention/csrc/flash_attn
Jeremy Reizenstein 0658e320f6
Preprocessor switches to control functionality (#788)
For faster and smaller builds in some simple cases,
provide switches to allow disabling
-backward
-alibi
-uneven k
-dropout
-local attention

Co-authored-by: Jeremy Francis Reizenstein <bottler@users.noreply.github.com>
2024-01-29 20:44:23 -08:00
..
src Preprocessor switches to control functionality (#788) 2024-01-29 20:44:23 -08:00
flash_api.cpp Preprocessor switches to control functionality (#788) 2024-01-29 20:44:23 -08:00