duanjunwen 8b37323f16 [feat] add run_fwd_bwd_with_microbatch (replace input) & test; add p&p.grad assert close test & all pass;		2024-08-27 09:31:38 +00:00
..
test_pipeline_utils	[shardformer] refactor pipeline grad ckpt config (#5646)	2024-04-25 15:19:30 +08:00
test_schedule	[feat] add run_fwd_bwd_with_microbatch (replace input) & test; add p&p.grad assert close test & all pass;	2024-08-27 09:31:38 +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