You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/tests/test_pipeline
Hongxin Liu 554aa9592e
[legacy] move communication and nn to legacy and refactor logger (#4671)
1 year ago
..
test_pipeline_utils [shardformer] Pipeline/whisper (#4456) 1 year ago
test_schedule [pipeline] set optimizer to optional in execute_pipeline (#4630) 1 year ago
rpc_test_utils.py
test_cuda_rpc_chimera.py
test_cuda_rpc_optimizer.py
test_cuda_rpc_pipeline.py
test_cuda_rpc_value_correctness.py
test_middleware_1f1b.py
test_p2p_communication.py [pipeline] implement p2p communication (#4100) 1 year ago
test_pipelinable.py [gemini] improve compatibility and add static placement policy (#4479) 1 year ago
test_pipeline_process_group.py
test_stage_manager.py [shardformer] support interleaved pipeline (#4448) 1 year ago