wassname/vllm
1
0
Fork 0
mirror of https://github.com/wassname/vllm.git synced 2026-07-12 23:14:17 +08:00
Code Issues Packages Projects Releases Wiki Activity
82 Commits 1 Branch 0 Tags
e9d3f2ff7772c8efe41dc805cec71c223ec18ec8
Commit Graph

3 Commits

Author SHA1 Message Date
Woosuk Kwon 3e9f991d6a Use FlashAttention for multi_query_kv_attention (#4) 2023-03-01 21:13:08 -08:00
Woosuk Kwon 0deacbce6e Implement single_query_cached_kv_attention kernel (#3) 2023-03-01 15:02:19 -08:00
Woosuk Kwon af68ec1c5c Add tests for kernels 2023-02-18 19:23:07 +00:00