ColossalAI/tests/test_fx/test_tracer/test_torchrec_model
..
test_deepfm_model.py
test_dlrm_model.py