提交 ed3b75b3 authored 作者: Frederic's avatar Frederic

Code cleanup

上级 4ceae0cd
......@@ -2020,7 +2020,7 @@ class GpuCAReduce(GpuOp):
}
""" % locals(), file=sio)
if self.reduce_mask == (0, 1, 0) or self.reduce_mask == (1, 0) or self.reduce_mask == (1, 0, 0):
if self.reduce_mask in [(0, 1, 0), (1, 0), (1, 0, 0)]:
reduce_fct = self._assign_reduce(node, nodename, "myresult",
"X[a * sX0 + b * sX1 + c * sX2]",
{}, True)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论