Merge branch 'sum' of https://github.com/zeroRains/Paddle into sum

PaddlePaddle · Apr 9, 2024 · c7ffb0a · c7ffb0a
2 parents 21157bc + 9fc3081
commit c7ffb0a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/paddle/phi/kernels/funcs/common_shape.h b/paddle/phi/kernels/funcs/common_shape.h
@@ -309,8 +309,8 @@ inline std::vector<int64_t> GetReduceDims(const DenseTensor &in,
       reduce_dims.push_back(i + diff);
     } else {
       PADDLE_ENFORCE_EQ(
-          in_dims[i],
-          out_dims[i + diff],
+          in_dims[i + diff],
+          out_dims[i],
           phi::errors::InvalidArgument(
               "ReduceDims dimension mismatch. Operands could "
               "not be broadcast together with the shape of in_dims = [%s] and "