提交 52721907 authored 作者: Frederic's avatar Frederic

change error type

上级 588f8a4b
...@@ -1047,7 +1047,9 @@ if cuda_available: ...@@ -1047,7 +1047,9 @@ if cuda_available:
def apply(self, fgraph): def apply(self, fgraph):
""" Raise a RuntimeError if cudnn can't be used""" """ Raise a RuntimeError if cudnn can't be used"""
if not dnn_available(): if not dnn_available():
raise RuntimeError( # Make an assert error as we want Theano to fail, not
# just skip this optimization.
raise AssertionError(
"cuDNN optimization was enabled, but Theano was not able" "cuDNN optimization was enabled, but Theano was not able"
" to use it. We got this error: \n" + " to use it. We got this error: \n" +
dnn_available.msg) dnn_available.msg)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论