提交 9a0e71c0 authored 作者: Frederic's avatar Frederic

skip cudnn test if we can't use it.

上级 eeb7c04a
...@@ -202,6 +202,8 @@ class TestDnnInferShapes(utt.InferShapeTester): ...@@ -202,6 +202,8 @@ class TestDnnInferShapes(utt.InferShapeTester):
self.mode = mode_with_gpu self.mode = mode_with_gpu
def test_softmax(self): def test_softmax(self):
if not dnn.dnn_available():
raise SkipTest(dnn.dnn_available.msg)
t = T.ftensor4('t') t = T.ftensor4('t')
rand_tensor = numpy.asarray( rand_tensor = numpy.asarray(
numpy.random.rand(5, 4, 3, 2), numpy.random.rand(5, 4, 3, 2),
...@@ -231,6 +233,8 @@ class TestDnnInferShapes(utt.InferShapeTester): ...@@ -231,6 +233,8 @@ class TestDnnInferShapes(utt.InferShapeTester):
) )
def test_conv(self): def test_conv(self):
if not dnn.dnn_available():
raise SkipTest(dnn.dnn_available.msg)
img = T.ftensor4('img') img = T.ftensor4('img')
kerns = T.ftensor4('kerns') kerns = T.ftensor4('kerns')
img_val = numpy.asarray( img_val = numpy.asarray(
...@@ -261,6 +265,8 @@ class TestDnnInferShapes(utt.InferShapeTester): ...@@ -261,6 +265,8 @@ class TestDnnInferShapes(utt.InferShapeTester):
) )
def test_conv_gradw(self): def test_conv_gradw(self):
if not dnn.dnn_available():
raise SkipTest(dnn.dnn_available.msg)
img = T.ftensor4('img') img = T.ftensor4('img')
kerns = T.ftensor4('kerns') kerns = T.ftensor4('kerns')
img_val = numpy.asarray( img_val = numpy.asarray(
...@@ -307,6 +313,8 @@ class TestDnnInferShapes(utt.InferShapeTester): ...@@ -307,6 +313,8 @@ class TestDnnInferShapes(utt.InferShapeTester):
) )
def test_conv_gradi(self): def test_conv_gradi(self):
if not dnn.dnn_available():
raise SkipTest(dnn.dnn_available.msg)
img = T.ftensor4('img') img = T.ftensor4('img')
kerns = T.ftensor4('kerns') kerns = T.ftensor4('kerns')
img_val = numpy.asarray( img_val = numpy.asarray(
...@@ -350,6 +358,8 @@ class TestDnnInferShapes(utt.InferShapeTester): ...@@ -350,6 +358,8 @@ class TestDnnInferShapes(utt.InferShapeTester):
) )
def test_pool(self): def test_pool(self):
if not dnn.dnn_available():
raise SkipTest(dnn.dnn_available.msg)
img = T.ftensor4('img') img = T.ftensor4('img')
img_val = numpy.asarray( img_val = numpy.asarray(
numpy.random.rand(2, 3, 4, 5), numpy.random.rand(2, 3, 4, 5),
...@@ -373,6 +383,8 @@ class TestDnnInferShapes(utt.InferShapeTester): ...@@ -373,6 +383,8 @@ class TestDnnInferShapes(utt.InferShapeTester):
) )
def test_pool_grad(self): def test_pool_grad(self):
if not dnn.dnn_available():
raise SkipTest(dnn.dnn_available.msg)
img = T.ftensor4('img') img = T.ftensor4('img')
img_grad = T.ftensor4('img_grad') img_grad = T.ftensor4('img_grad')
out = T.ftensor4('out') out = T.ftensor4('out')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论