frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +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_ops_gradients.py
View File

@ -14,6 +14,7 @@ from torch.testing._internal.common_utils import (
TestCase,
TestGradients,
unMarkDynamoStrictTest,
xfailIfS390X,
)
from torch.testing._internal.custom_op_db import custom_op_db
from torch.testing._internal.hop_db import hop_db
@ -28,6 +29,7 @@ _gradcheck_ops = partial(
@unMarkDynamoStrictTest
class TestBwdGradients(TestGradients):
# Tests that gradients are computed correctly
@xfailIfS390X
@_gradcheck_ops(op_db + hop_db + custom_op_db)
def test_fn_grad(self, device, dtype, op):
# This is verified by test_dtypes in test_ops.py