flash-attention/csrc/flash_attn
2023-10-08 16:26:33 -07:00
..
src Change constexpr int to constexpr static int 2023-10-08 16:26:33 -07:00
flash_api.cpp [Gen] Accept cache_batch_idx to index into the KV cache 2023-10-03 16:27:26 -07:00