squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
e279bf8ed9
flash-attention/csrc/flash_attn
History
Tri Dao e279bf8ed9 [Gen] Accept cache_batch_idx to index into the KV cache
2023-10-03 16:27:26 -07:00
..
src [Gen] Accept cache_batch_idx to index into the KV cache 2023-10-03 16:27:26 -07:00
flash_api.cpp [Gen] Accept cache_batch_idx to index into the KV cache 2023-10-03 16:27:26 -07:00