github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/pipeline
History
Kirigaya Kazuto 170fa81095
[pipeline/chimera] test chimera | fix bug of initializing (#1615) 
* [pipeline/tuning] improve dispatch performance both time and space cost

* [pipeline/converge] add interface for testing convergence

* [NFC] polish colossalai/utils/multi_tensor_apply/multi_tensor_apply.py code style

* Update PipelineBase.py

* [pipeline/chimera] reconstruct PipelineBase and Worker to support more feasible custom schedule | finish Chimera

* [pipeline/chimera] test chimera | fix bug of initializing
 		2022-09-20 18:00:39 +08:00
..
rpc [pipeline/chimera] test chimera | fix bug of initializing (#1615) 2022-09-20 18:00:39 +08:00
__init__.py [pipeline] refactor the pipeline module (#1087) 2022-06-10 11:27:38 +08:00
layer_sepc.py [pipeline] refactor the pipeline module (#1087) 2022-06-10 11:27:38 +08:00
pipelinable.py [pipeline/chimera] test chimera | fix bug of initializing (#1615) 2022-09-20 18:00:39 +08:00
pipeline_process_group.py [pipeline/chimera] test chimera | fix bug of initializing (#1615) 2022-09-20 18:00:39 +08:00
utils.py [NFC] polish colossalai/pipeline/utils.py code style (#1562) 2022-09-08 22:11:04 +08:00