提交 6f549791 authored 作者: carriepl's avatar carriepl

Fix typo

上级 00821f97
......@@ -1478,7 +1478,7 @@ def local_log_softmax_dnn(node):
@register_opt('cudnn')
@op_lifter([LogSoftmax])
def local_logsoftmax_to_dnn(node, ctx_name):
if not dnn_available(ctx_name) or version() < 3:
if not dnn_available(ctx_name) or version() < 3000:
# No log-softmax before cudnn v3
return
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论