biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity
5,124 Commits 2 Branches 140 Tags
8a4a2efc6fc32cdc30e4e35ba3f8c64dcd0aa1d0
Commit Graph

253 Commits

Author SHA1 Message Date
Woosuk Kwon
a1b3de86cd Refactor the test code for attention kernels (#13) 2023-03-29 18:59:27 -07:00
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