提交 b83b1c15 authored 作者: Ian Goodfellow's avatar Ian Goodfellow

marked _k_init as safe for general ops

上级 8624c2f9
...@@ -819,7 +819,6 @@ class GpuCAReduce(GpuOp): ...@@ -819,7 +819,6 @@ class GpuCAReduce(GpuOp):
return sio.getvalue() return sio.getvalue()
def _k_init(self, *args): def _k_init(self, *args):
self._op_guard()
return """ return """
const int threadCount = blockDim.x * blockDim.y * blockDim.z; const int threadCount = blockDim.x * blockDim.y * blockDim.z;
const int threadNum = threadIdx.z * blockDim.x * blockDim.y + threadIdx.y * blockDim.x + threadIdx.x; const int threadNum = threadIdx.z * blockDim.x * blockDim.y + threadIdx.y * blockDim.x + threadIdx.x;
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论