Files
ColossalAI/tests/test_context/configs/parallel_2d_init.py