OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Fixed an issue with XPU skip so the test_decompose_mem_bound_mm.py suite can be ran correctly (#153245)

Browse Source 
Fixes #153239

Replaced custom decorator with the common one. Although the better way to skip the whole suite would be to add it to skip list in run_test.py

Pull Request resolved: https://github.com/pytorch/pytorch/pull/153245
Approved by: https://github.com/jeffdaily
This commit is contained in:
iupaikov-amd 2025-05-27 23:10:21 +00:00 committed by PyTorch MergeBot
parent 4b39832412
commit 3f10c9d8af
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
test/inductor/test_decompose_mem_bound_mm.py
View File

@ -1,6 +1,7 @@
# Owner(s): ["module: inductor"] # Owner(s): ["module: inductor"]
import logging import logging
import unittest
import torch import torch
import torch._inductor import torch._inductor
@ -12,7 +13,7 @@ from torch.testing import FileCheck
from torch.testing._internal.common_utils import ( from torch.testing._internal.common_utils import (
instantiate_parametrized_tests, instantiate_parametrized_tests,
parametrize, parametrize,
skipIfXpu, TEST_XPU,
) )
from torch.testing._internal.inductor_utils import GPU_TYPE, HAS_CUDA from torch.testing._internal.inductor_utils import GPU_TYPE, HAS_CUDA
from torch.testing._internal.triton_utils import requires_gpu from torch.testing._internal.triton_utils import requires_gpu
@ -48,9 +49,10 @@ class MyModule3(torch.nn.Module):
@requires_gpu @requires_gpu
@skipIfXpu( @unittest.skipIf(
msg="Intel GPU has not enabled decompose_mem_bound_mm PASS in " TEST_XPU,
"torch/_inductor/fx_passes/decompose_mem_bound_mm.py" "Intel GPU has not enabled decompose_mem_bound_mm PASS in "
"torch/_inductor/fx_passes/decompose_mem_bound_mm.py",
) )
@torch._inductor.config.patch( @torch._inductor.config.patch(
post_grad_fusion_options={ post_grad_fusion_options={
@ -276,6 +278,8 @@ class TestDecomposeMemMM(TestCase):
) )
counters.clear() counters.clear()
# (1, 64, 32, False) vesrion fails
@unittest.skip
@parametrize( @parametrize(
"m,k,n, should_decompose", "m,k,n, should_decompose",
[(1, 64, 16, True), (2, 64, 16, False), (1, 64, 32, False)], [(1, 64, 16, True), (2, 64, 16, False), (1, 64, 32, False)],
@ -338,6 +342,7 @@ class TestDecomposeMemMM(TestCase):
) )
counters.clear() counters.clear()
@unittest.skip
@parametrize("m,k,n, should_decompose", [(20480, 5, 2, True)]) @parametrize("m,k,n, should_decompose", [(20480, 5, 2, True)])
@parametrize("has_bias", [True, False]) @parametrize("has_bias", [True, False])
def test_dynamic_shape(self, m, n, k, has_bias, should_decompose): def test_dynamic_shape(self, m, n, k, has_bias, should_decompose):