[NPU] replace ReduceSumD with ReduceSum (#1277)

PaddlePaddle · Jun 4, 2024 · 2444df5 · 2444df5
1 parent c3c84d5
commit 2444df5
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/backends/npu/kernels/elementwise_mul_kernel.cc b/backends/npu/kernels/elementwise_mul_kernel.cc
@@ -39,8 +39,12 @@ static void ReduceDims(const Context& dev_ctx,
     }
   }
   dev_ctx.template Alloc<T>(out);
-  const auto& runner = NpuOpRunner(
-      "ReduceSumD", {in}, {*out}, {{"axes", axes}, {"keep_dims", false}});
+  NpuOpRunner runner;
+  runner.SetType("ReduceSum");
+  runner.AddInput(in);
+  runner.AddInput(dev_ctx, std::move(axes));
+  runner.AddOutput(*out);
+  runner.AddAttr("keep_dims", false);
   runner.Run(stream);
 }