提交 2708e558 authored 作者: Frederic's avatar Frederic

Fix warning reported on the mailing list

上级 6f01cf86
...@@ -1718,11 +1718,11 @@ if True: ...@@ -1718,11 +1718,11 @@ if True:
@register_opt('cudnn') @register_opt('cudnn')
@local_optimizer([SoftmaxGrad]) @local_optimizer([SoftmaxGrad])
def local_softmax_dnn_grad(node): def local_softmax_dnn_grad(node):
if ( if (isinstance(node.op, SoftmaxGrad) and
isinstance(node.op, SoftmaxGrad) ((node.inputs[0].owner and
and (isinstance(node.inputs[0].owner.op, HostFromGpu) isinstance(node.inputs[0].owner.op, HostFromGpu))
or isinstance(node.inputs[1].owner.op, HostFromGpu)) or (node.inputs[1].owner and
): isinstance(node.inputs[1].owner.op, HostFromGpu)))):
if not dnn_available(): if not dnn_available():
return return
ins = [] ins = []
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论