提交 a1304650 authored 作者: Frederic Bastien's avatar Frederic Bastien

Fix DebugMode when a perform raise NotImplementedError. This happen with this…

Fix DebugMode when a perform raise NotImplementedError. This happen with this test theano/sandbox/cuda/tests/test_conv_cuda_ndarray.py:test_conv_grads
上级 721666e2
......@@ -2043,7 +2043,7 @@ class _Linker(gof.link.LocalLinker):
"output storage", i)
try:
thunk_py()
except utils.MethodNotDefined:
except (utils.MethodNotDefined, NotImplementedError):
# shouldn't have put it into the list in
# the first place
thunk_py = None
......
......@@ -306,7 +306,7 @@ class AbstractConv2d(BaseAbstractConv2d):
raise NotImplementedError(
'AbstractConv2d theano optimization failed. '
'Did you exclude both "conv_dnn" and "conv_gemm" from '
'the optimizer?')
'the optimizer? Is cudnn available and does the GPU support it?')
def grad(self, inp, grads):
bottom, weights = inp
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论