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_interleaved.py	[Feature] Zigzag Ring attention (#5905)	2024-08-16 13:56:38 +08:00
test_oneF_oneB.py	[Feature] Zigzag Ring attention (#5905)	2024-08-16 13:56:38 +08:00
test_pipeline_schedule_utils.py	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
test_zerobubble_poc.py	[feat] add test run_fwd_bwd automatic scheduling;	2024-08-26 11:21:56 +00:00
test_zerobubble_pp.py	[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