提交 7d181357 authored 作者: Olivier Breuleux's avatar Olivier Breuleux

corrected CAReduce if argument is a scalar

上级 0e361fb5
......@@ -603,10 +603,13 @@ class CAReduce(Op):
}
""" % locals()
if len(axis) == 1:
all_code = [("", "")] * nnested + [(task0_decl, code1), ""]
if nnested:
if len(axis) == 1:
all_code = [("", "")] * nnested + [(task0_decl, code1), ""]
else:
all_code = [("", "")] * nnested + [(task0_decl, "")] + [("", "")] * (len(axis) - 2) + [("", code1), ""]
else:
all_code = [("", "")] * nnested + [(task0_decl, "")] + [("", "")] * (len(axis) - 2) + [("", code1), ""]
all_code = [task0_decl + code1]
loop = cgen.make_loop([order, range(nnested) + ['x'] * len(axis)], [idtype, odtype], all_code, sub)
return decl, checks, alloc, loop
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论