fixed a small bug

f5a19acf · Shucai Xiao · f54be162 · f5a19acf
Commit f5a19acf authored Jul 01, 2019 by Shucai Xiao
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/targets/gpu/device/include/migraphx/gpu/device/reduce.hpp ...targets/gpu/device/include/migraphx/gpu/device/reduce.hpp +1 -1

No files found.
--- a/src/targets/gpu/device/include/migraphx/gpu/device/reduce.hpp
+++ b/src/targets/gpu/device/include/migraphx/gpu/device/reduce.hpp
@@ -182,7 +182,7 @@ __device__ auto block_reduce(index idx, Op op, T init, std::size_t n, F f)
    }
    __syncthreads();
-    type y = 0;
+    type y = init;
    for(std::size_t i = 0; i < idx.nlocal() / 64; i++)
    {
        y = op(y, buffer[i]);