mirror of https://github.com/hpcaitech/ColossalAI
![]() * add kvcache manager funcs for batching * add batch bucket for batching * revise RunningList struct in handler * add kvcache/batch funcs for compatibility * use new batching methods * fix indexing bugs * revise abort logic * use cpu seq lengths/block tables * rm unused attr in Sequence * fix type conversion/default arg * add and revise pytests * revise pytests, rm unused tests * rm unused statements * fix pop finished indexing issue * fix: use index in batch when retrieving inputs/update seqs * use dict instead of odict in batch struct * arg type hinting * fix make compress * refine comments * fix: pop_n_seqs to pop the first n seqs * add check in request handler * remove redundant conversion * fix test for request handler * fix pop method in batch bucket * fix prefill adding |
||
---|---|---|
.. | ||
test_models | ||
test_ops/triton | ||
_utils.py | ||
test_batch_bucket.py | ||
test_config_and_struct.py | ||
test_inference_engine.py | ||
test_kvcache_manager.py | ||
test_request_handler.py |