flash-attention/csrc/flash_attn
Jeremy Reizenstein ce3e7280f8
Allow varlen_fwd to take optional seqused_k (#647)
Co-authored-by: bottler <bottler@users.noreply.github.com>
2023-11-27 00:41:23 -08:00
..
src Allow varlen_fwd to take optional seqused_k (#647) 2023-11-27 00:41:23 -08:00
flash_api.cpp Allow varlen_fwd to take optional seqused_k (#647) 2023-11-27 00:41:23 -08:00