提交 0a7415d7 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Merge pull request #3289 from abergeron/disable_softmaxgrad_v3

Disable the optimization to use DnnSoftmaxGrad for cuDNN v3 since it …
......@@ -2305,7 +2305,8 @@ if True:
isinstance(node.inputs[0].owner.op, HostFromGpu)) or
(node.inputs[1].owner and
isinstance(node.inputs[1].owner.op, HostFromGpu)))):
if not dnn_available():
if not dnn_available() or version() != (2000, 2000):
# Softmax grad is broken in v3 rc1 for this case
return
ins = []
for n in node.inputs:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论