mirror of https://github.com/hpcaitech/ColossalAI
0df5034a36
* [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 * [pipeline/pytree] add pytree to process args and kwargs | provide to process args and kwargs after forward * [pipeline/fix-bug] num_microbatches support any integrate | stable chimera | launch tool for rpc pp framework |
||
---|---|---|
.. | ||
rpc | ||
__init__.py | ||
layer_sepc.py | ||
pipelinable.py | ||
pipeline_process_group.py | ||
utils.py |