Commit 30b15ebc authored by Shucai Xiao's avatar Shucai Xiao
Browse files

revert some chagnes.

parent ec5a90ae
...@@ -78,9 +78,9 @@ __device__ auto block_reduce(index idx, Op op, T init, std::size_t n, F f) ...@@ -78,9 +78,9 @@ __device__ auto block_reduce(index idx, Op op, T init, std::size_t n, F f)
for(std::size_t s = 1; s < idx.nlocal(); s *= 2) for(std::size_t s = 1; s < idx.nlocal(); s *= 2)
{ {
const std::size_t index = 2 * s * idx.local; const std::size_t index = 2 * s * idx.local;
if(index < idx.nlocal()) if(index + s < idx.nlocal())
{ {
buffer[index + s] = op(buffer[index], buffer[index + s]); buffer[index] = op(buffer[index], buffer[index + s]);
} }
__syncthreads(); __syncthreads();
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment