ColossalAI/tests/test_pipeline/test_schedule
duanjunwen e6e1a97a6d [fix] fix requir grad position and detach position and input&output local buffer append position; 2024-09-04 03:31:08 +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_pp.py [fix] fix requir grad position and detach position and input&output local buffer append position; 2024-09-04 03:31:08 +00:00