saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[Inductor UT][Intel GPU] Skip test case which doesn't currently work on the XPU stack but newly re-enabled by community. (#127629)

Browse source 
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
This commit is contained in:
xinan.lin 2024-05-31 10:28:11 -07:00 committed by PyTorch MergeBot
parent 2cef2fc2b4
commit c19ad112f6
1 changed files with 3 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
test/inductor/test_triton_heuristics.py
View file

@ -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()