ColossalAI/colossalai/pipeline
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
layer_sepc.py
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