ColossalAI/tests/test_infer/test_dynamic_batching
Jianghai e0757c31fb
[inference] Dynamic Batching for Single and Multiple GPUs (#4831)
* finish batch manager

* 1

* first

* fix

* fix dynamic batching

* llama infer

* finish test

* support different lengths generating

* del prints

* del prints

* fix

* fix bug

---------

Co-authored-by: CjhHa1 <cjh18671720497outlook.com>
2023-10-11 17:52:52 +08:00
..
test_dynamic_batching_manager.py [inference] Dynamic Batching for Single and Multiple GPUs (#4831) 2023-10-11 17:52:52 +08:00
test_forward.py [inference] Dynamic Batching for Single and Multiple GPUs (#4831) 2023-10-11 17:52:52 +08:00