William Wen
2023-10-28 04:14:53 +00:00
committed by PyTorch MergeBot
parent 31f605344f
commit a380bf3297
3 changed files with 13 additions and 0 deletions

View File

@ -17,6 +17,7 @@ from torch.testing._internal.common_utils import (
skipIfTorchDynamo,
suppress_warnings,
TEST_WITH_ASAN,
TEST_WITH_TORCHDYNAMO,
run_tests,
dtype_abbrs,
parametrize
@ -1105,6 +1106,13 @@ class TestMeta(TestCase):
@suppress_warnings
@ops(op_db)
def test_meta_outplace(self, device, dtype, op):
skip_op_names = (
"fft.ihfft",
"fft.ihfft2",
"linalg.lu_solve",
)
if TEST_WITH_TORCHDYNAMO and op.name in skip_op_names:
raise unittest.SkipTest("flaky")
# run the OpInfo sample inputs, cross-referencing them with the
# meta implementation and check the results are the same. All
# the heavy lifting happens in MetaCrossRefFunctionMode