tests/test_pipeline/test_cuda_rpc_optimizer.py · edc9e419ad5a4c91aeed8c0be7915e66b7490aac · OpenDAS / ColossalAI

[pipeline/chimera] reconstruct PipelineBase and Worker to support more... · edc9e419

Kirigaya Kazuto authored Sep 19, 2022

[pipeline/chimera] reconstruct PipelineBase and Worker to support more feasible custom schedule | finish Chimera (#1595)

* [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

edc9e419

test_cuda_rpc_optimizer.py 2.42 KB

Replace test_cuda_rpc_optimizer.py