ColossalAI/tests/test_pipeline/test_schedule
duanjunwen a5ec3d4285 [fix] fix mem; use a new model shape; only assert mem less and equal than theo; 2024-09-09 06:38:31 +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 mem; use a new model shape; only assert mem less and equal than theo; 2024-09-09 06:38:31 +00:00