提交 3009ff45 authored 作者: --global's avatar --global

Skip test_pooling3d if cudnn not recent enough

上级 b094c0bb
...@@ -323,7 +323,9 @@ def test_pooling(): ...@@ -323,7 +323,9 @@ def test_pooling():
def test_pooling3d(): def test_pooling3d():
if not cuda.dnn.dnn_available(): # CuDNN 3d pooling requires CuDNN v3. Don't test if the CuDNN version is
# too old.
if not cuda.dnn.dnn_available() or cuda.dnn.version() < (3000, 3000):
raise SkipTest(cuda.dnn.dnn_available.msg) raise SkipTest(cuda.dnn.dnn_available.msg)
x = T.TensorType(broadcastable=(False, False, False, False, False), x = T.TensorType(broadcastable=(False, False, False, False, False),
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论