pytorch - 💡(How to fix) Fix Unify the definition and usage of skipOps [1 participants]

# New skipOps decorator
def skipOps(to_skip):
    def wrapped(fn):
        fn._skipops = tuple(to_skip)
        return fn
    return wrapped

# Update ops class to read test._skipops
class ops(_TestParametrizer):
    def _parametrize_test(self, test, generic_cls, device_cls):
        self.update_skip_ops(test)  # Reads test._skipops and appends it to the decorator of the corresponding operation.
        ...

# Example migration of @skipOps callers
# Before
@suppress_warnings
@ops(_op_db, allowed_dtypes=(torch.float,))
@skipOps(
    _op_db,
    "TestLocalDTensorOps",
    "test_dtensor_op_db",
    dtensor_fails | dtensor_fails_no_strategy,
)
def test_dtensor_op_db(self, dtype, op):
    ...

# After
@suppress_warnings
@ops(_op_db, allowed_dtypes=(torch.float,))
@skipOps(
    dtensor_fails | dtensor_fails_no_strategy,
)
def test_dtensor_op_db(self, dtype, op):
    ...