wassname/vllm
1
0
Fork 0
mirror of https://github.com/wassname/vllm.git synced 2026-07-17 11:34:17 +08:00
Code Issues Packages Projects Releases Wiki Activity
149 Commits 1 Branch 0 Tags
057daef778ec4e951841f44afda1cd0b1eb50ee4
Commit Graph
10 Commits
Author SHA1 Message Date
Woosuk KwonandGitHub e070829ae8 Support bfloat16 data type (#54) 2023-05-03 14:09:44 -07:00
Siyuan (Ryans) ZhuangandGitHub e3cec88aa5 Memcpy kernel for flash attention (#29) 
* optimize

* add benchmark

* add assert

* add test
 2023-04-10 18:22:49 -07:00
Woosuk KwonandGitHub 0f40557af6 Implement block copy kernel to optimize beam search (#32) 2023-04-07 17:45:07 -07:00
Woosuk KwonandGitHub 897cb2ae28 Optimize data movement (#20) 2023-04-02 00:30:17 -07:00
Woosuk KwonandGitHub 88c0268a18 Implement custom kernel for LLaMA rotary embedding (#14) 2023-03-30 11:04:21 -07:00
Woosuk KwonandGitHub cfae35b861 Add miscellaneous updates (#8) 2023-03-13 13:48:38 -07:00
Woosuk KwonandGitHub 1a7eb7da61 Support beam search & parallel generation (#7) 2023-03-10 09:58:21 -08:00
Woosuk KwonandGitHub 0deacbce6e Implement single_query_cached_kv_attention kernel (#3) 2023-03-01 15:02:19 -08:00
Woosuk Kwon c413c41cda Add reshape_and_cache op 2023-02-18 19:22:57 +00:00
Woosuk Kwon ffad4e1e03 cache_kernel -> cache_kernels 2023-02-16 20:05:45 +00:00