ColossalAI/tests/test_trainer
..
test_pipeline
test_trainer_with_non_pipe_schedule.py
test_trainer_with_pipe_schedule.py