提交 ebb7f5ca 编写于 作者: M Megvii Engine Team

fix(imperative/python/test): add device judgment for the test case of conv-transpose2d-quantized

GitOrigin-RevId: 456cb78439e5f08fd562541e58e045b9a162ba34
上级 66c18f60
......@@ -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)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册