Skip some unittests (#90609)

* Skip a unittest that needs FFT if not built with FFT
* Mark a test with "slow": `python test/test_ops.py -k TestCompositeComplianceCUDA.test_forward_ad_svd_lowrank_cuda_float32` took >5min on my machine.
* Skip a flaky test that's marked "expectedFailure", similar to #90233
Pull Request resolved: https://github.com/pytorch/pytorch/pull/90609
Approved by: https://github.com/soumith
This commit is contained in:
Yuxin Wu
2022-12-11 23:53:02 +00:00
committed by PyTorch MergeBot
parent 11442accc6
commit 8127724c3b
2 changed files with 5 additions and 2 deletions

View File

@ -4,6 +4,7 @@ import os
import sys
import torch
from torch.utils._pytree import tree_map
import unittest
from torch.testing._internal.common_utils import run_tests
from torch.fx.operator_schemas import normalize_function
@ -229,6 +230,7 @@ class TestSchemaCheck(JitTestCase):
self.assertEqual(expected, actual)
# Tests that SchemaCheckMode wraps torch.Tensor when there is a kwarg tensor input
@unittest.skipIf(not torch._C.has_spectral, "ATen not built with FFT.")
def test_schema_check_mode_functionality_kwarg_tensor(self):
x = torch.rand((3, 5))
w = torch.rand((4))