ColossalAI/tests/test_pipeline
duanjunwen 6af81d8c0d [feat] add fwd_bwd_step, run_fwd_only; 2024-08-30 02:47:52 +00:00
..
test_pipeline_utils [shardformer] refactor pipeline grad ckpt config (#5646) 2024-04-25 15:19:30 +08:00
test_schedule [feat] add fwd_bwd_step, run_fwd_only; 2024-08-30 02:47:52 +00:00
test_p2p_communication.py [Feature] optimize PP overlap (#5735) 2024-06-26 14:48:02 +08:00
test_stage_manager.py [Feature] optimize PP overlap (#5735) 2024-06-26 14:48:02 +08:00