From 494aac65a78d3105d8a125c155e8795734538a01 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Fri, 24 Jun 2022 19:35:00 +0200 Subject: [PATCH] Skip `test_multi_gpu_data_parallel_forward` for `MaskFormer` (#17864) Co-authored-by: ydshieh --- tests/models/maskformer/test_modeling_maskformer.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/tests/models/maskformer/test_modeling_maskformer.py b/tests/models/maskformer/test_modeling_maskformer.py index bbc24719d7..1c64ca46a5 100644 --- a/tests/models/maskformer/test_modeling_maskformer.py +++ b/tests/models/maskformer/test_modeling_maskformer.py @@ -21,7 +21,7 @@ import numpy as np from tests.test_modeling_common import floats_tensor from transformers import DetrConfig, MaskFormerConfig, SwinConfig, is_torch_available, is_vision_available -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 from ...test_configuration_common import ConfigTester @@ -212,6 +212,13 @@ class MaskFormerModelTest(ModelTesterMixin, unittest.TestCase): def test_resize_tokens_embeddings(self): pass + @require_torch_multi_gpu + @unittest.skip( + reason="MaskFormer has some layers using `add_module` which doesn't work well with `nn.DataParallel`" + ) + def test_multi_gpu_data_parallel_forward(self): + pass + def test_forward_signature(self): config, _ = self.model_tester.prepare_config_and_inputs_for_common()