Files
ColossalAI/tests/test_ops/test_loss_func.py