Woosuk KwonandGitHub 0f40557af6 Implement block copy kernel to optimize beam search (#32)

2023-04-07 17:45:07 -07:00

..

http_frontend

Fix potential bugs in FastAPI frontend and add comments (#28)

2023-04-06 13:44:24 +08:00

master

Add CUDA graph-based all reduce launcher (#26)

2023-04-05 11:16:57 -07:00

models

Implement block copy kernel to optimize beam search (#32)

2023-04-07 17:45:07 -07:00

parallel_utils

Add CUDA graph-based all reduce launcher (#26)

2023-04-05 11:16:57 -07:00

worker

Implement block copy kernel to optimize beam search (#32)

2023-04-07 17:45:07 -07:00

block.py

Support beam search & parallel generation (#7)

2023-03-10 09:58:21 -08:00

sampling_params.py

FastAPI-based working frontend (#10)

2023-03-29 14:48:56 +08:00

sequence.py

Implement preemption via recomputation & Refactor scheduling logic (#12)

2023-03-30 14:51:46 -07:00

utils.py

FastAPI-based working frontend (#10)

2023-03-29 14:48:56 +08:00