biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity
13,168 Commits 2 Branches 140 Tags
019e2c3b7ca799e4497a67eb7e49edea361cb90e
Commit Graph

4203 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