Files
ColossalAI/tests/test_device/test_device_mesh.py