提交 0956a9fb authored 作者: James Bergstra's avatar James Bergstra

GpuSum - fix error in case 010

上级 3b4aaa54
......@@ -1195,6 +1195,7 @@ class GpuSum(Op):
""" %locals()
def c_code_cache_version(self):
return ()
return (15,)
......@@ -1352,9 +1353,9 @@ class GpuSum(Op):
return; //TODO: set error code
}
for (int i0 = threadIdx.x; i0 < d0; i0 += blockDim.x)
for (int i1 = threadIdx.x; i1 < d1; i1 += blockDim.x)
{
mysum += A[blockIdx.x * sA0 + i0 * sA1 + blockIdx.y * sA2];
mysum += A[blockIdx.x * sA0 + i1 * sA1 + blockIdx.y * sA2];
}
%(reducebuf)s
}
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论