提交 81cd1db0 authored 作者: Olivier Breuleux's avatar Olivier Breuleux

oops! I forgot to commit this bugfix...

上级 aa00203b
...@@ -603,7 +603,7 @@ class CAReduce(Op): ...@@ -603,7 +603,7 @@ class CAReduce(Op):
} }
""" % locals() """ % locals()
if nnested: if node.inputs[0].type.ndim:
if len(axis) == 1: if len(axis) == 1:
all_code = [("", "")] * nnested + [(task0_decl, code1), ""] all_code = [("", "")] * nnested + [(task0_decl, code1), ""]
else: else:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论