github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-06 19:40:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

[hotfix] fix testcase in test_fx/test_tracer (#5779)

Browse Source 
* [fix] branch for fix testcase;

* [fix] fix test_analyzer & test_auto_parallel;

* [fix] remove local change about moe;

* [fix] rm local change moe;

* [fix] fix test_deepfm_model & test_dlrf_model；

* [fix] fix test_hf_albert & test_hf_gpt;
This commit is contained in:
duanjunwen
2024-06-05 11:29:32 +08:00
committed by GitHub
parent 80c3c8789b
commit 10a19e22c6
4 changed files with 9 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_fx/test_tracer/test_torchrec_model/test_deepfm_model.py
View File

@@ -52,7 +52,7 @@ def trace_and_compare(model_cls, data, output_transform_fn, meta_args=None):
@clear_cache_before_run()
def test_torchrec_deepfm_models():
deepfm_models = model_zoo.get_sub_registry("deepfm")
deepfm_models = model_zoo.get_sub_registry(keyword="deepfm", allow_empty=True)
torch.backends.cudnn.deterministic = True
for name, (model_fn, data_gen_fn, output_transform_fn, _, attribute) in deepfm_models.items():

2
tests/test_fx/test_tracer/test_torchrec_model/test_dlrm_model.py
View File

@@ -53,7 +53,7 @@ def trace_and_compare(model_cls, data, output_transform_fn, meta_args=None):
@clear_cache_before_run()
def test_torchrec_dlrm_models():
torch.backends.cudnn.deterministic = True
dlrm_models = model_zoo.get_sub_registry("dlrm")
dlrm_models = model_zoo.get_sub_registry(keyword="deepfm", allow_empty=True)
for name, (model_fn, data_gen_fn, output_transform_fn, _, attribute) in dlrm_models.items():
data = data_gen_fn()