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

[CI] Add inductor cpu accuracy test running on AVX2 runners (#128682)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/128682
Approved by: https://github.com/jgong5, https://github.com/desertfire
This commit is contained in:
zengxian
2024-07-26 13:24:41 +00:00
committed by PyTorch MergeBot
parent e73fa28ec8
commit d3e932dc10
4 changed files with 19 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_ops.py
View File

@ -254,6 +254,7 @@ class TestCommon(TestCase):
# This test runs in double and complex double precision because
# NumPy does computation internally using double precision for many functions
# resulting in possible equality check failures.
# skip windows case on CPU due to https://github.com/pytorch/pytorch/issues/129947
@onlyNativeDeviceTypesAnd(["hpu"])
@suppress_warnings
@ops(_ref_test_ops, allowed_dtypes=(torch.float64, torch.long, torch.complex128))
@ -264,6 +265,7 @@ class TestCommon(TestCase):
in ("signal_windows_exponential", "signal_windows_bartlett")
and dtype == torch.float64
and "cuda" in device
or "cpu" in device
): # noqa: E121
raise unittest.SkipTest("XXX: raises tensor-likes are not close.")