提交 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): ...@@ -2043,7 +2043,7 @@ class _Linker(gof.link.LocalLinker):
"output storage", i) "output storage", i)
try: try:
thunk_py() thunk_py()
except utils.MethodNotDefined: except (utils.MethodNotDefined, NotImplementedError):
# shouldn't have put it into the list in # shouldn't have put it into the list in
# the first place # the first place
thunk_py = None thunk_py = None
......
...@@ -306,7 +306,7 @@ class AbstractConv2d(BaseAbstractConv2d): ...@@ -306,7 +306,7 @@ class AbstractConv2d(BaseAbstractConv2d):
raise NotImplementedError( raise NotImplementedError(
'AbstractConv2d theano optimization failed. ' 'AbstractConv2d theano optimization failed. '
'Did you exclude both "conv_dnn" and "conv_gemm" from ' '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): def grad(self, inp, grads):
bottom, weights = inp bottom, weights = inp
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论