提交 349378f1 authored 作者: --global's avatar --global

Skip certain tests when using CuDNN with a version older than v3

上级 32ad87b4
...@@ -521,9 +521,16 @@ def _test_full(cls, mode=None, version=[-1], extra_shapes=[], ...@@ -521,9 +521,16 @@ def _test_full(cls, mode=None, version=[-1], extra_shapes=[],
def test_full(): def test_full():
for t in _test_full(None,
mode=theano_mode, # If using CuDNN version before v3, only run the tests where the
version=[-1]): # kernels are not larger than the input in any spatial dimension.
if cuda.dnn.dnn_available() and cuda.dnn.version() < (3000, 3000):
test_bigger_kernels = False
else:
test_bigger_kernels = True
for t in _test_full(None, mode=theano_mode, version=[-1],
test_bigger_kernels=test_bigger_kernels):
yield t yield t
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论