From 8f76dc8e5aaad58f2df7748b6d6970376f315a9a Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Mon, 15 May 2023 16:46:58 +0200 Subject: [PATCH] Skip failing `AlignModelTest::test_multi_gpu_data_parallel_forward` (#23374) fix Co-authored-by: ydshieh --- tests/models/align/test_modeling_align.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/models/align/test_modeling_align.py b/tests/models/align/test_modeling_align.py index 02d046924c..2357c20e21 100644 --- a/tests/models/align/test_modeling_align.py +++ b/tests/models/align/test_modeling_align.py @@ -437,6 +437,10 @@ class AlignModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase): config_and_inputs = self.model_tester.prepare_config_and_inputs() self.model_tester.create_and_check_model(*config_and_inputs) + @unittest.skip(reason="Start to fail after using torch `cu118`.") + def test_multi_gpu_data_parallel_forward(self): + super().test_multi_gpu_data_parallel_forward() + @unittest.skip(reason="Hidden_states is tested in individual model tests") def test_hidden_states_output(self): pass