diff --git a/imperative/python/test/unit/quantization/test_op.py b/imperative/python/test/unit/quantization/test_op.py index 9b07ac02ae098b89ac785779d35173c4edd02a1a..11934ae5149eb6729333444223afbab61b6c4b48 100644 --- a/imperative/python/test/unit/quantization/test_op.py +++ b/imperative/python/test/unit/quantization/test_op.py @@ -171,6 +171,10 @@ def test_conv_bias(): run(10, 36, 8, 46, 26, 2, 2, 2, 1, 1, 2, True, "relu") +@pytest.mark.skipif( + get_device_count("gpu") > 0, + reason="does not support int8 when gpu compute capability less than 6.1", +) def test_conv_transpose2d(): rng = np.random.RandomState(seed=2021)