ColossalAI/tests/test_fx/test_tracer/test_torchrec_model
Frank Lee 30e50c8b4a
[autoparallel] implemented all matmul strategy generator (#1650)
2022-09-27 12:06:25 +08:00
..
test_deepfm_model.py [autoparallel] implemented all matmul strategy generator (#1650) 2022-09-27 12:06:25 +08:00
test_dlrm_model.py [autoparallel] implemented all matmul strategy generator (#1650) 2022-09-27 12:06:25 +08:00