Format

924a0ffa · Paul · 73d53f86 · 924a0ffa
Commit 924a0ffa authored Aug 19, 2022 by Paul
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 5 deletions

src/targets/gpu/kernels/include/migraphx/kernels/reduce.hpp src/targets/gpu/kernels/include/migraphx/kernels/reduce.hpp +3 -5

No files found.
--- a/src/targets/gpu/kernels/include/migraphx/kernels/reduce.hpp
+++ b/src/targets/gpu/kernels/include/migraphx/kernels/reduce.hpp
@@ -201,11 +201,9 @@ struct block
        __device__ auto reduce(Op op, T init, Read read) const
        {
            return sliced(slicer, [=](auto x, auto... xs) {
-                return block_reduce(idx,
+                return block_reduce(idx, op, init, x.get_shape().elements(), [&](auto j) {
-                                               op,
+                    return vec_reduce(read(x[j], xs[j]...), op);
-                                               init,
+                });
-                                               x.get_shape().elements(),
-                                            [&](auto j) { return vec_reduce(read(x[j], xs[j]...), op); });
            });
        }