ColossalAI/tests/test_pipeline
Kirigaya Kazuto 6159d45417
[pipeline/tuning] improve dispatch performance both time and space cost (#1544)
2022-09-07 19:01:06 +08:00
..
rpc_test_utils.py [pipeline/tuning] improve dispatch performance both time and space cost (#1544) 2022-09-07 19:01:06 +08:00
test_cuda_rpc_optimizer.py [pipeline/tuning] improve dispatch performance both time and space cost (#1544) 2022-09-07 19:01:06 +08:00
test_cuda_rpc_pipeline.py [pipeline/rpc] update outstanding mechanism | optimize dispatching strategy (#1497) 2022-08-26 14:04:23 +08:00
test_cuda_rpc_value_correctness.py [pipeline/rpc] update outstanding mechanism | optimize dispatching strategy (#1497) 2022-08-26 14:04:23 +08:00
test_pipelinable.py
test_pipeline_process_group.py [pipeline/pipleline_process_group] finish PipelineProcessGroup to manage local abd global rank in TP,DP and PP (#1508) 2022-09-01 17:45:47 +08:00