[NPU] fix reduce_max (#43230)

07ede118 · Aganlengzi · GitHub · 3adeea60 · 07ede118
隐藏空白更改
内联并排

Showing with 3 addition and 0 deletion

paddle/fluid/operators/reduce_ops/reduce_max_op_npu.cc paddle/fluid/operators/reduce_ops/reduce_max_op_npu.cc +3 -0

未找到文件。
--- a/paddle/fluid/operators/reduce_ops/reduce_max_op_npu.cc
+++ b/paddle/fluid/operators/reduce_ops/reduce_max_op_npu.cc
@@ -141,6 +141,9 @@ class ReduceMaxGradNPUKernel : public framework::OpKernel<T> {
    Tensor tmp_out, tmp_out_grad;
    auto tmp_out_dims_vec = x_dims_vec;
    for (auto d : reduce_dims) {
+      if (d < 0) {
+        d += x_dims_vec.size();
+      }
      tmp_out_dims_vec[d] = 1;
    }