提交 9af9b162 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

skip cudnn test if not available

上级 674f8d51
......@@ -511,6 +511,8 @@ def test_pooling_opt_arbitrary_dimensions():
def test_pooling_empty_batch():
if not dnn.dnn_available(test_ctx_name):
raise SkipTest(dnn.dnn_available.msg)
img_shp = (0, 5, 6, 8)
img = T.ftensor4('img')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论