fix bug in bit_count and moe_gate_permute. (#73539)

liuruyan · web-flow · commit 02b3495378fd · 2025-06-23T10:37:00.000+08:00
diff --git a/paddle/phi/infermeta/ternary.cc b/paddle/phi/infermeta/ternary.cc
@@ -1817,16 +1817,6 @@ void MoeGateDispatchPermuteInferMeta(const MetaTensor& x,
             "The dimensions of Input(corr_bias) must be 1, but received "
             "dimensions of Input(corr_bias) is [%d]",
             corr_bias_dims.size()));
-    PADDLE_ENFORCE_EQ(
-        corr_bias_dims[0],
-        x_dims[0],
-        common::errors::InvalidArgument(
-            "The dimensions of Input(corr_bias) must be equal to the first "
-            "dimension of Input(x), but received Input(corr_bias) first "
-            "dimension is [%d],"
-            "Input(x) first dimension is [%d]",
-            corr_bias_dims[0],
-            x_dims[0]));
     PADDLE_ENFORCE_EQ(
         corr_bias.dtype(),
         paddle::DataType::FLOAT32,
diff --git a/paddle/phi/infermeta/unary.cc b/paddle/phi/infermeta/unary.cc
@@ -6487,17 +6487,11 @@ void IntBincountInferMeta(const MetaTensor& x,
                           int64_t high,
                           int64_t dtype,
                           MetaTensor* out) {
-  PADDLE_ENFORCE_EQ(
-      x.dims().size(),
-      1,
-      errors::InvalidArgument(
-          "The input 'x' of int_bincount must be a 1-D Tensor, but got %u-D.",
-          x.dims().size()));
   PADDLE_ENFORCE_GT(
       high,
       low,
       errors::InvalidArgument("Attr high (%d) must be > low (%d).", high, low));
-  int64_t bin_count = high - low + 1;
+  int64_t bin_count = high - low;
 
   out->set_dims(phi::make_ddim({bin_count}));
   out->set_dtype(x.dtype());
