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

Run only listed tests on s390x (#140265)

Browse Source 
Skip tests that are failing

This was previously part of https://github.com/pytorch/pytorch/pull/125401

Pull Request resolved: https://github.com/pytorch/pytorch/pull/140265
Approved by: https://github.com/malfet

Co-authored-by: Nikita Shulga <2453524+malfet@users.noreply.github.com>
This commit is contained in:
Aleksei Nikiforov
2024-11-20 22:53:09 +00:00
committed by PyTorch MergeBot
parent 701e06b643
commit a82bab6419
13 changed files with 398 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_optim.py
View File

@ -44,6 +44,7 @@ from torch.testing._internal.common_utils import (
run_tests,
TEST_WITH_TORCHDYNAMO,
TestCase,
xfailIfS390X,
)
@ -580,6 +581,7 @@ class TestOptimRenewed(TestCase):
self.assertEqual(complex_steps, real_steps)
@skipMPS
@xfailIfS390X
@optims([o for o in optim_db if o.supports_complex], dtypes=[torch.complex64])
def test_complex_2d(self, device, dtype, optim_info):
optim_cls = optim_info.optim_cls