Files
ColossalAI/tests/test_fx/test_pipeline_passes.py