Skip to content

Commit 667939a

Browse files
authored
[tests] add the missing require_torch_multi_gpu flag (#30250)
add gpu flag
1 parent 440bd3c commit 667939a

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/models/bros/test_modeling_bros.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
import copy
1818
import unittest
1919

20-
from transformers.testing_utils import require_torch, slow, torch_device
20+
from transformers.testing_utils import require_torch, require_torch_multi_gpu, slow, torch_device
2121
from transformers.utils import is_torch_available
2222

2323
from ...test_configuration_common import ConfigTester
@@ -344,6 +344,7 @@ def test_model(self):
344344
config_and_inputs = self.model_tester.prepare_config_and_inputs()
345345
self.model_tester.create_and_check_model(*config_and_inputs)
346346

347+
@require_torch_multi_gpu
347348
def test_multi_gpu_data_parallel_forward(self):
348349
super().test_multi_gpu_data_parallel_forward()
349350

0 commit comments

Comments
 (0)