提交 9783d9a9 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Disable the optimization to use DnnSoftmaxGrad for cuDNN v3 since it is broken…

Disable the optimization to use DnnSoftmaxGrad for cuDNN v3 since it is broken for (n, c 1, 1) shapes at the moment.
上级 15c90dd3
...@@ -2280,7 +2280,8 @@ if True: ...@@ -2280,7 +2280,8 @@ if True:
isinstance(node.inputs[0].owner.op, HostFromGpu)) or isinstance(node.inputs[0].owner.op, HostFromGpu)) or
(node.inputs[1].owner and (node.inputs[1].owner and
isinstance(node.inputs[1].owner.op, HostFromGpu)))): 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 return
ins = [] ins = []
for n in node.inputs: for n in node.inputs:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论