Files
ColossalAI/tests/test_amp