From 5176dc2310c6659f8e4c4b90252ab585e6445ba6 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Mon, 5 Jun 2023 16:29:32 +0200 Subject: [PATCH] Skip `test_multi_gpu_data_parallel_forward` for `MobileViTV2ModelTest` (#24017) * fix * fix * fix --------- Co-authored-by: ydshieh --- tests/models/mobilevitv2/test_modeling_mobilevitv2.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/tests/models/mobilevitv2/test_modeling_mobilevitv2.py b/tests/models/mobilevitv2/test_modeling_mobilevitv2.py index 0da4afd375..8b3a754d33 100644 --- a/tests/models/mobilevitv2/test_modeling_mobilevitv2.py +++ b/tests/models/mobilevitv2/test_modeling_mobilevitv2.py @@ -19,7 +19,7 @@ import inspect import unittest from transformers import MobileViTV2Config -from transformers.testing_utils import require_torch, require_vision, slow, torch_device +from transformers.testing_utils import require_torch, require_torch_multi_gpu, require_vision, slow, torch_device from transformers.utils import cached_property, is_torch_available, is_vision_available from ...test_configuration_common import ConfigTester @@ -220,6 +220,11 @@ class MobileViTV2ModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestC def test_attention_outputs(self): pass + @require_torch_multi_gpu + @unittest.skip(reason="Got `CUDA error: misaligned address` for tests after this one being run.") + def test_multi_gpu_data_parallel_forward(self): + pass + def test_forward_signature(self): config, _ = self.model_tester.prepare_config_and_inputs_for_common()