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

Enable more nightly tests on s390x (#148452)

Browse Source 
Also enable some tests which probably were accidentally disabled.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/148452
Approved by: https://github.com/seemethere, https://github.com/malfet
This commit is contained in:
Aleksei Nikiforov
2025-03-18 16:09:35 +00:00
committed by PyTorch MergeBot
parent 381d0cb239
commit d5b1d99f78
13 changed files with 391 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_optim.py
View File

@ -45,7 +45,6 @@ from torch.testing._internal.common_utils import (
run_tests,
TEST_WITH_TORCHDYNAMO,
TestCase,
xfailIfS390X,
)
@ -591,7 +590,6 @@ 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