提交 367aedf8 authored 作者: Frederic's avatar Frederic

don't check version when not needed

上级 ccb84dc5
...@@ -2388,12 +2388,8 @@ if True: ...@@ -2388,12 +2388,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)))):
v = version()
if v[0] != v[1]:
return
if not dnn_available(): if not dnn_available():
# Softmax grad is broken in v3 rc1 for this case # Softmax grad is broken in v3 rc1 for this case
# But we don't support cudnn v3 rc version now.
return return
ins = [] ins = []
for n in node.inputs: for n in node.inputs:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论