change shape_analysis to infer_context

Author: yulangz

paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/multiary_infer_sym.cc

@@ -505,20 +505,20 @@ bool MemoryEfficientAttentionOpInferSymbolicShape(
 }
 
 bool RoiAlignOpInferSymbolicShape(
-    pir::Operation *op, pir::ShapeConstraintIRAnalysis *shape_analysis) {
+    pir::Operation *op, pir::InferSymbolicShapeContext *infer_context) {
   const auto &x = op->operand_source(0);
   const auto &boxes = op->operand_source(1);
 
   const auto &num_boxes =
-      shape_analysis->GetShapeOrDataForValue(boxes).shape()[0];
+      infer_context->GetShapeOrDataForValue(boxes).shape()[0];
   symbol::DimExpr channel_num =
-      shape_analysis->GetShapeOrDataForValue(x).shape()[1];
+      infer_context->GetShapeOrDataForValue(x).shape()[1];
 
   int32_t out_h = op->attribute<pir::Int32Attribute>("pooled_height").data();
   int32_t out_w = op->attribute<pir::Int32Attribute>("pooled_width").data();
 
   std::vector<symbol::DimExpr> out_dim = {num_boxes, channel_num, out_h, out_w};
-  shape_analysis->SetShapeOrDataForValue(
+  infer_context->SetShapeOrDataForValue(
       op->result(0), symbol::TensorShapeOrDataDimExprs(out_dim));
   return true;
 }

paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/unary_infer_sym.cc

@@ -247,7 +247,7 @@ bool DiagonalOpInferSymbolicShape(
 }
 
 bool DistributeFpnProposalsOpInferSymbolicShape(
-    pir::Operation *op, pir::ShapeConstraintIRAnalysis *shape_analysis) {
+    pir::Operation *op, pir::InferSymbolicShapeContext *infer_context) {
   const auto &attributes = op->attributes();
   int32_t min_level =
       attributes.at("min_level").dyn_cast<pir::Int32Attribute>().data();
@@ -258,7 +258,7 @@ bool DistributeFpnProposalsOpInferSymbolicShape(
   symbol::DimExpr num_rois = [&]() {
     pir::Value rois_num = op->operand_source(1);
     const auto &rois_num_shape_or_data =
-        shape_analysis->GetShapeOrDataForValue(rois_num);
+        infer_context->GetShapeOrDataForValue(rois_num);
 
     PADDLE_ENFORCE_EQ(
         rois_num_shape_or_data.shape()[0],
@@ -290,7 +290,7 @@ bool DistributeFpnProposalsOpInferSymbolicShape(
     } else {
       symbol::DimExpr last_dim = num_rois;
       for (int i = 0; i < num_levels - 1; i++) {
-        const auto &next_sym_name = shape_analysis->GetNextSymName();
+        const auto &next_sym_name = infer_context->GetNextSymName();
         std::vector<symbol::DimExpr> level_dim = {next_sym_name, 4};
         multi_rois_out_shape.emplace_back(
             symbol::TensorShapeOrDataDimExprs(level_dim));
@@ -314,15 +314,15 @@ bool DistributeFpnProposalsOpInferSymbolicShape(
     return symbol::TensorShapeOrDataDimExprs({num_rois, 1});
   }();
 
-  shape_analysis->SetShapeOrDataForValue(op->result(0), multi_rois_out_shape);
-  shape_analysis->SetShapeOrDataForValue(op->result(1),
-                                         rois_num_per_level_out_shape);
-  shape_analysis->SetShapeOrDataForValue(op->result(2), restore_ind);
+  infer_context->SetShapeOrDataForValue(op->result(0), multi_rois_out_shape);
+  infer_context->SetShapeOrDataForValue(op->result(1),
+                                        rois_num_per_level_out_shape);
+  infer_context->SetShapeOrDataForValue(op->result(2), restore_ind);
   return true;
 }
 
-bool EinsumOpInferSymbolicShape(
-    pir::Operation *op, pir::ShapeConstraintIRAnalysis *shape_analysis) {
+bool EinsumOpInferSymbolicShape(pir::Operation *op,
+                                pir::InferSymbolicShapeContext *infer_context) {
   PADDLE_THROW(phi::errors::Unimplemented(
       op->name() + " 's InferSymbolicShape interface is NOT implemented now."));
   return true;