提交 21f95489 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Fix error message when not available

上级 d66a7734
...@@ -178,7 +178,7 @@ class TestConv2d(unittest.TestCase): ...@@ -178,7 +178,7 @@ class TestConv2d(unittest.TestCase):
def test_dnn_conv(self): def test_dnn_conv(self):
if not dnn_available(test_ctx_name): if not dnn_available(test_ctx_name):
raise SkipTest(cuda.dnn.dnn_available.msg) raise SkipTest(dnn_available.msg)
mode = mode_with_gpu mode = mode_with_gpu
# provide_shape is not used by the CuDNN impementation # provide_shape is not used by the CuDNN impementation
provide_shape = False provide_shape = False
...@@ -206,7 +206,7 @@ class TestConv2d(unittest.TestCase): ...@@ -206,7 +206,7 @@ class TestConv2d(unittest.TestCase):
def test_cormm_conv(self): def test_cormm_conv(self):
if not dnn_available(test_ctx_name): if not dnn_available(test_ctx_name):
raise SkipTest(cuda.dnn.dnn_available.msg) raise SkipTest(dnn_available.msg)
mode = mode_without_gpu mode = mode_without_gpu
for (i, f), s, b, flip, provide_shape in itertools.product( for (i, f), s, b, flip, provide_shape in itertools.product(
...@@ -234,7 +234,7 @@ class TestConv2d(unittest.TestCase): ...@@ -234,7 +234,7 @@ class TestConv2d(unittest.TestCase):
def test_cpu_conv(self): def test_cpu_conv(self):
if not dnn_available(test_ctx_name): if not dnn_available(test_ctx_name):
raise SkipTest(cuda.dnn.dnn_available.msg) raise SkipTest(dnn_available.msg)
mode = mode_without_gpu.excluding('conv_gemm') mode = mode_without_gpu.excluding('conv_gemm')
for (i, f), s, b, flip, provide_shape in itertools.product( for (i, f), s, b, flip, provide_shape in itertools.product(
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论