frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[BE][Ez]: Unskip conv1d MPS test (#154795)

Browse Source 
Fixes issue I noticed where conv1d test is skipped for complex types unconditionally
Pull Request resolved: https://github.com/pytorch/pytorch/pull/154795
Approved by: https://github.com/jansel
This commit is contained in:
Aaron Gokaslan
2025-05-31 23:01:16 +00:00
committed by PyTorch MergeBot
parent b85c460749
commit dbad6d71c7
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/nn/test_convolution.py
View File

@ -1797,7 +1797,9 @@ class TestConvolutionNNDeviceType(NNTestCase):
self.assertEqual(expect, actual)
@dtypes(torch.float, torch.cfloat)
@dtypesIfMPS(torch.float)
@dtypesIfMPS(
*([torch.float] if MACOS_VERSION < 14.0 else [torch.float, torch.cfloat])
) # Complex not supported on MacOS13
def test_conv1d_same_padding_backward(self, device, dtype):
# Test F.conv1d gradients work with padding='same'
x = torch.rand(1, 1, 12, dtype=dtype, device=device, requires_grad=True)