ColossalAI/tests/test_infer
Frank Lee db1a763307
[inference] removed redundancy init_batch (#5353)
2024-02-02 11:44:15 +08:00
..
test_models [Infer] Optimize Blocked KVCache And Kernels Using It (#5325) 2024-01-30 16:06:09 +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] removed redundancy init_batch (#5353) 2024-02-02 11:44:15 +08:00
test_inference_engine.py [inference] simplified config verification (#5346) 2024-02-01 15:31:01 +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