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

GpuSum - fix error in case 010

上级 3b4aaa54
...@@ -1195,6 +1195,7 @@ class GpuSum(Op): ...@@ -1195,6 +1195,7 @@ class GpuSum(Op):
""" %locals() """ %locals()
def c_code_cache_version(self): def c_code_cache_version(self):
return ()
return (15,) return (15,)
...@@ -1352,9 +1353,9 @@ class GpuSum(Op): ...@@ -1352,9 +1353,9 @@ class GpuSum(Op):
return; //TODO: set error code 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 %(reducebuf)s
} }
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论