From fe140464213b60dbc5dafa772fa13f0809e3d8b4 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Thu, 30 Jun 2022 18:05:29 +0200 Subject: [PATCH] skip some ipex tests until it works with torch 1.12 (#17964) Co-authored-by: ydshieh --- tests/trainer/test_trainer.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tests/trainer/test_trainer.py b/tests/trainer/test_trainer.py index 9b1aa40016..c2d435e257 100644 --- a/tests/trainer/test_trainer.py +++ b/tests/trainer/test_trainer.py @@ -642,6 +642,7 @@ class TrainerIntegrationTest(TestCasePlus, TrainerIntegrationCommon): train_output = trainer.train() self.assertEqual(train_output.global_step, 10) + @unittest.skip(reason="skip temporarily until intel_extension_for_pytorch works with torch 1.12") @require_torch_bf16_cpu @require_intel_extension_for_pytorch def test_number_of_steps_in_training_with_ipex(self): @@ -886,6 +887,7 @@ class TrainerIntegrationTest(TestCasePlus, TrainerIntegrationCommon): expected_acc = AlmostAccuracy()((pred + 1, y))["accuracy"] self.assertAlmostEqual(results["eval_accuracy"], expected_acc) + @unittest.skip(reason="skip temporarily until intel_extension_for_pytorch works with torch 1.12") @require_torch_bf16_cpu @require_intel_extension_for_pytorch def test_evaluate_with_ipex(self): @@ -1006,6 +1008,7 @@ class TrainerIntegrationTest(TestCasePlus, TrainerIntegrationCommon): self.assertTrue(np.array_equal(labels[0], trainer.eval_dataset.ys[0])) self.assertTrue(np.array_equal(labels[1], trainer.eval_dataset.ys[1])) + @unittest.skip(reason="skip temporarily until intel_extension_for_pytorch works with torch 1.12") @require_torch_bf16_cpu @require_intel_extension_for_pytorch def test_predict_with_ipex(self):