From ebb7f5cac23ebe8209772c8cb46296bd5348f9bb Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Tue, 10 Aug 2021 17:39:11 +0800 Subject: [PATCH] fix(imperative/python/test): add device judgment for the test case of conv-transpose2d-quantized GitOrigin-RevId: 456cb78439e5f08fd562541e58e045b9a162ba34 --- imperative/python/test/unit/quantization/test_op.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/imperative/python/test/unit/quantization/test_op.py b/imperative/python/test/unit/quantization/test_op.py index 9b07ac02..11934ae5 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)