mirror of https://github.com/hpcaitech/ColossalAI
170fa81095
* [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 |
||
|---|---|---|
| .. | ||
| rpc_test_utils.py | ||
| test_cuda_rpc_chimera.py | ||
| test_cuda_rpc_optimizer.py | ||
| test_cuda_rpc_performance.py | ||
| test_cuda_rpc_pipeline.py | ||
| test_cuda_rpc_value_correctness.py | ||
| test_pipelinable.py | ||
| test_pipeline_process_group.py | ||