From efe264b7518bbd036e64ac4470f8abf1e38acda8 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Mon, 22 Mar 2021 11:43:28 +0800 Subject: [PATCH] test(mge/utils): skip network node convbias test in gpu GitOrigin-RevId: 9d06cf6c5ad4c75840b5efb95278b8a7d989b73f --- imperative/python/test/unit/utils/test_network_node.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/imperative/python/test/unit/utils/test_network_node.py b/imperative/python/test/unit/utils/test_network_node.py index 62896bad..d293c3c4 100644 --- a/imperative/python/test/unit/utils/test_network_node.py +++ b/imperative/python/test/unit/utils/test_network_node.py @@ -14,6 +14,7 @@ from megengine.core._imperative_rt.core2 import apply from megengine.core._wrap import Device from megengine.core.ops import builtin from megengine.device import is_cuda_available +from megengine.distributed.helper import get_device_count_by_fork from megengine.functional.external import tensorrt_runtime_opr from megengine.jit.tracing import trace from megengine.tensor import Tensor @@ -265,6 +266,10 @@ def test_deformable_ps_roi_pooling(): check_pygraph_dump(fwd, [inp, rois, trans], [result]) +@pytest.mark.skipif( + get_device_count_by_fork("gpu") > 0, + reason="does not support int8 when gpu compute capability less than 6.1", +) def test_convbias(): @trace(symbolic=True, capture_as_const=True) def fwd(inp, weight, bias): -- GitLab