From c19ad112f65ab88af3eeb319dda0675c076b0b67 Mon Sep 17 00:00:00 2001 From: "xinan.lin" Date: Fri, 31 May 2024 10:28:11 -0700 Subject: [PATCH] [Inductor UT][Intel GPU] Skip test case which doesn't currently work on the XPU stack but newly re-enabled by community. (#127629) The Inductor UT test/inductor/test_triton_heuristics.py:test_artificial_zgrid that previously skipped was recently enbaled by the PR https://github.com/pytorch/pytorch/pull/127448. However, the test doesn't currently work on the XPU stack, it will huang on GPU, so this PR skip the test for Intel GPU instead of expected failure. Pull Request resolved: https://github.com/pytorch/pytorch/pull/127629 Approved by: https://github.com/EikanWang, https://github.com/peterbell10 --- test/inductor/test_triton_heuristics.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/test/inductor/test_triton_heuristics.py b/test/inductor/test_triton_heuristics.py index c0908251f85..549903f47ce 100644 --- a/test/inductor/test_triton_heuristics.py +++ b/test/inductor/test_triton_heuristics.py @@ -4,9 +4,8 @@ import sys import unittest import torch -from torch.testing._internal.common_device_type import expectedFailureXPU -from torch.testing._internal.common_utils import IS_LINUX +from torch.testing._internal.common_utils import IS_LINUX, skipIfXpu from torch.testing._internal.inductor_utils import GPU_TYPE, HAS_GPU try: @@ -73,11 +72,11 @@ class TestTritonHeuristics(TestCase): ] self.assertEqual(forward(*args), foo_c(*args)) - @expectedFailureXPU + @skipIfXpu def test_artificial_zgrid(self): self._test_artificial_zgrid() - @expectedFailureXPU + @skipIfXpu @config.patch("cpp_wrapper", True) def test_artificial_grid_cpp_wrapper(self): self._test_artificial_zgrid()