You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

Bin Jia 1db6727678 [Pipeline inference] Combine kvcache with pipeline inference (#4938) ... * merge kvcache with pipeline inference and refactor the code structure * support ppsize > 2 * refactor pipeline code * do pre-commit * modify benchmark * fix bench mark * polish code * add docstring and update readme * refactor the code * fix some logic bug of ppinfer * polish readme * fix typo * skip infer test		1 year ago
..
_C	…
_analyzer	[misc] update pre-commit and run all files (#4752)	1 year ago
amp	[feature] Add clip_grad_norm for hybrid_parallel_plugin (#4837)	1 year ago
auto_parallel	[misc] update pre-commit and run all files (#4752)	1 year ago
autochunk	[misc] update pre-commit and run all files (#4752)	1 year ago
booster	[gemini] support gradient accumulation (#4869)	1 year ago
checkpoint_io	[checkpointio] hotfix torch 2.0 compatibility (#4824)	1 year ago
cli	[bug] Fix the version check bug in colossalai run when generating the cmd. (#4713)	1 year ago
cluster	[doc] polish shardformer doc (#4779)	1 year ago
context	[misc] update pre-commit and run all files (#4752)	1 year ago
device	[misc] update pre-commit and run all files (#4752)	1 year ago
fx	[misc] update pre-commit and run all files (#4752)	1 year ago
inference	[Pipeline inference] Combine kvcache with pipeline inference (#4938)	1 year ago
interface	[lazy] support from_pretrained (#4801)	1 year ago
kernel	[inference] add reference and fix some bugs (#4937)	1 year ago
lazy	[doc] add lazy init docs (#4808)	1 year ago
legacy	[hotfix] fix torch 2.0 compatibility (#4936)	1 year ago
logging	[misc] update pre-commit and run all files (#4752)	1 year ago
nn	[test] add no master test for low level zero plugin (#4934)	1 year ago
pipeline	[Pipeline inference] Combine kvcache with pipeline inference (#4938)	1 year ago
shardformer	[Pipeline inference] Combine kvcache with pipeline inference (#4938)	1 year ago
tensor	[misc] update pre-commit and run all files (#4752)	1 year ago
testing	[test] merge old components to test to model zoo (#4945)	1 year ago
utils	[misc] update pre-commit and run all files (#4752)	1 year ago
zero	[gemini] support gradient accumulation (#4869)	1 year ago
__init__.py	[misc] update pre-commit and run all files (#4752)	1 year ago
initialize.py	[misc] update pre-commit and run all files (#4752)	1 year ago