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

corrected CAReduce if argument is a scalar

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