ColossalAI/tests/test_context
..
configs
test_hybrid_parallel.py