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

fix gh-2262. Fix crash cause by using cudnn when we shouldn't

上级 4ad5236b
...@@ -1164,6 +1164,8 @@ if cuda_available: ...@@ -1164,6 +1164,8 @@ if cuda_available:
and (isinstance(node.inputs[0].owner.op, HostFromGpu) and (isinstance(node.inputs[0].owner.op, HostFromGpu)
or isinstance(node.inputs[1].owner.op, HostFromGpu)) or isinstance(node.inputs[1].owner.op, HostFromGpu))
): ):
if not dnn_available():
return
ins = [] ins = []
for n in node.inputs: for n in node.inputs:
if isinstance(n.owner.op, HostFromGpu): if isinstance(n.owner.op, HostFromGpu):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论