integral_types_and, all_types, double_types)
from .._core import _dispatch_dtypes
from torch.testing._internal.common_device_type import \
- (onlyOnCPUAndCUDA, skipIf, skipCUDAIfNoMagma, skipCUDAIfNoMagmaAndNoCusolver, skipCUDAIfNoCusolver,
+ (expectedFailure, onlyOnCPUAndCUDA, skipIf, skipCUDAIfNoMagma, skipCUDAIfNoMagmaAndNoCusolver, skipCUDAIfNoCusolver,
skipCPUIfNoLapack, skipCPUIfNoFFT, skipCUDAIfRocm, precisionOverride, toleranceOverride, tol)
from torch.testing._internal.common_cuda import CUDA11OrLater, SM53OrLater, SM60OrLater
from torch.testing._internal.common_utils import \
an operator. Any test that matches all provided arguments will be skipped.
The skip will only be checked if the active_if argument is True."""
- def __init__(self, cls_name=None, test_name=None, *,
- device_type=None, dtypes=None, active_if=True):
- super().__init__(decorators=skipIf(True, "Skipped!"), cls_name=cls_name,
- test_name=test_name, device_type=device_type, dtypes=dtypes,
- active_if=active_if)
+ def __init__(
+ self, cls_name=None, test_name=None, *, device_type=None, dtypes=None, active_if=True,
+ expected_failure=False):
+ """
+ Args:
+ cls_name: the name of the test class to skip
+ test_name: the name of the test within the test class to skip
+ device_type: the devices for which to skip the tests
+ dtypes: the dtypes for which to skip the tests
+ active_if: whether tests matching the above arguments should be skipped
+ expected_failure: whether to assert that skipped tests fail
+ """
+ decorator = expectedFailure(device_type) if expected_failure else skipIf(True, "Skipped!")
+ super().__init__(decorators=decorator, cls_name=cls_name, test_name=test_name,
+ device_type=device_type, dtypes=dtypes, active_if=active_if)
+
class SampleInput(object):
"""Represents sample inputs to a function."""