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

[MPS/inductor] Adjust more tests that depends on non-divisible input sizes (#147681)

Browse Source 
Also adjust a comment while I'm at it.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/147681
Approved by: https://github.com/jansel
This commit is contained in:
Davide Italiano 2025-02-23 00:33:24 +00:00 committed by PyTorch MergeBot
parent 6a5e3917a7
commit b084635735
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
test/inductor/test_torchinductor.py
View File

@ -4624,6 +4624,7 @@ class CommonTemplate:
) )
@skip_if_gpu_halide # slow @skip_if_gpu_halide # slow
@xfail_if_mps # Non-divisible input sizes are not implemented on MPS device
def test_adaptive_avg_pool2d1(self): def test_adaptive_avg_pool2d1(self):
def fn(x): def fn(x):
return aten._adaptive_avg_pool2d(x, (6, 6)), aten._adaptive_avg_pool2d( return aten._adaptive_avg_pool2d(x, (6, 6)), aten._adaptive_avg_pool2d(
@ -4648,6 +4649,7 @@ class CommonTemplate:
(torch.randn(2, 4, 6, 6),), (torch.randn(2, 4, 6, 6),),
) )
@xfail_if_mps # Non-divisible input sizes are not implemented on MPS device
def test_adaptive_avg_pool2d2(self): def test_adaptive_avg_pool2d2(self):
# Big kernel size, use fallback # Big kernel size, use fallback
def fn(x): def fn(x):
@ -4947,7 +4949,7 @@ class CommonTemplate:
self.assertEqual(eager_delta, compile_delta) self.assertEqual(eager_delta, compile_delta)
@xfail_if_mps @xfail_if_mps # Non-divisible input sizes are not implemented on MPS device
def test_adaptive_avg_pool_with_output_size_0(self): def test_adaptive_avg_pool_with_output_size_0(self):
m1 = nn.AdaptiveAvgPool1d(0) m1 = nn.AdaptiveAvgPool1d(0)
self.common(m1, (torch.randn(1, 2),)) self.common(m1, (torch.randn(1, 2),))