Files
ColossalAI/tests/test_context/test_hybrid_parallel.py