add check

Author: ooooo-create

paddle/fluid/pir/dialect/operator/ir/manual_op.cc

@@ -3588,6 +3588,12 @@ std::vector<pir::Type> ExpandOp::InferMeta(
                            .dyn_cast<paddle::dialect::IntArrayAttribute>()
                            .data()
                            .GetData();
+      PADDLE_ENFORCE_LE(shape_vec.size(),
+                        1,
+                        common::errors::InvalidArgument(
+                            "The size of shape for Full op should be less than "
+                            "or equal to 1, but receive %d.",
+                            shape_vec.size()));
       auto items = shape_vec.empty() ? 1 : shape_vec[0];
       vec_shape = std::vector<int64_t>(items, shape_item);
     } else if (shape.isa<pir::OpResult>() &&

paddle/fluid/pir/dialect/operator/utils/utils.cc

@@ -432,6 +432,12 @@ std::vector<int64_t> ParseValueShape(const pir::Value& shape,
                          .dyn_cast<paddle::dialect::IntArrayAttribute>()
                          .data()
                          .GetData();
+    PADDLE_ENFORCE_LE(shape_vec.size(),
+                      1,
+                      common::errors::InvalidArgument(
+                          "The size of shape for Full op should be less than "
+                          "or equal to 1, but receive %d.",
+                          shape_vec.size()));
     auto items = shape_vec.empty() ? 1 : shape_vec[0];
     vec_shape = std::vector<int64_t>(items, shape_item);
   } else if (shape.isa<pir::OpResult>() &&