提交 50bee243 authored 作者: --global's avatar --global

Adjust conditions for skipping tests in InferShapeTester

上级 c2703ca1
...@@ -725,8 +725,8 @@ class TestDnnInferShapes(utt.InferShapeTester): ...@@ -725,8 +725,8 @@ class TestDnnInferShapes(utt.InferShapeTester):
) )
def test_conv_gradi(self): def test_conv_gradi(self):
if not (cuda.dnn.dnn_available() and dnn.version() >= (2000, 2000)): if not dnn.dnn_available():
raise SkipTest('"CuDNN 3D convolution requires CuDNN v2') raise SkipTest(dnn.dnn_available.msg)
img = T.ftensor4('img') img = T.ftensor4('img')
kerns = T.ftensor4('kerns') kerns = T.ftensor4('kerns')
out = T.ftensor4('out') out = T.ftensor4('out')
...@@ -770,8 +770,8 @@ class TestDnnInferShapes(utt.InferShapeTester): ...@@ -770,8 +770,8 @@ class TestDnnInferShapes(utt.InferShapeTester):
) )
def test_conv3d_gradi(self): def test_conv3d_gradi(self):
if not dnn.dnn_available(): if not (cuda.dnn.dnn_available() and dnn.version() >= (2000, 2000)):
raise SkipTest(dnn.dnn_available.msg) raise SkipTest('"CuDNN 3D convolution requires CuDNN v2')
ftensor5 = T.TensorType(dtype="float32", broadcastable=(False,) * 5) ftensor5 = T.TensorType(dtype="float32", broadcastable=(False,) * 5)
img = ftensor5('img') img = ftensor5('img')
kerns = ftensor5('kerns') kerns = ftensor5('kerns')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论