SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Model parallel tests should return, not pass in non model parallel settings. (#8825)

Browse Source
This commit is contained in:
Lysandre Debut
2020-11-27 16:41:29 -05:00
committed by GitHub
parent edbff1fd00
commit 18c32eeb21
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_modeling_common.py
View File

@@ -1022,7 +1022,7 @@ class ModelTesterMixin:
@require_torch_multi_gpu @require_torch_multi_gpu
def test_model_parallelization(self): def test_model_parallelization(self):
if not self.test_model_parallel: if not self.test_model_parallel:
pass return
import subprocess import subprocess
@@ -1079,7 +1079,7 @@ class ModelTesterMixin:
@require_torch_multi_gpu @require_torch_multi_gpu
def test_model_parallel_equal_results(self): def test_model_parallel_equal_results(self):
if not self.test_model_parallel: if not self.test_model_parallel:
pass return
config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common() config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common()