ColossalAI/tests/test_infer
yuehuayingxueluo 631862f339
[Inference]Optimize generation process of inference engine (#5356)
* opt inference engine

* fix run_benchmark.sh

* fix generate in engine.py

* rollback tesh_inference_engine.py
2024-02-02 15:38:21 +08:00
..
test_models [Infer] Optimize Blocked KVCache And Kernels Using It (#5325) 2024-01-30 16:06:09 +08:00
test_ops/triton [inference] moved ops tests to test_infer (#5354) 2024-02-02 13:51:22 +08:00
_utils.py [Inference] Add the logic of the inference engine (#5173) 2024-01-11 13:39:56 +00:00
test_config_and_struct.py [inference] moved ops tests to test_infer (#5354) 2024-02-02 13:51:22 +08:00
test_inference_engine.py [Inference]Optimize generation process of inference engine (#5356) 2024-02-02 15:38:21 +08:00
test_kvcache_manager.py [Infer] Optimize Blocked KVCache And Kernels Using It (#5325) 2024-01-30 16:06:09 +08:00
test_request_handler.py [inference]Optimize the usage of the mid tensors space in flash attn (#5304) 2024-01-26 14:00:10 +08:00