fix merge bug (PaddlePaddle#64069)

co63oc · May 10, 2024 · 7ef02c0 · 7ef02c0
1 parent 5c5e46d
commit 7ef02c0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/nullary_infer_sym.cc b/paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/nullary_infer_sym.cc
@@ -294,15 +294,15 @@ bool GaussianOpInferSymbolicShape(
   } else {
     pir::Value operand_source = op->operand_source(0);
     const symbol::ShapeOrDataDimExprs &operand_shape_or_data =
-        shape_analysis->GetShapeOrDataForValue(operand_source);
+        infer_context->GetShapeOrDataForValue(operand_source);
     PADDLE_ENFORCE_EQ(
         operand_shape_or_data.data().has_value(),
         true,
         common::errors::InvalidArgument(
             "The data of input dim_expr shape is null. When input of empty op "
             "is a tensor, the data of input dim_expr shape must have value."));
 
-    shape_analysis->SetShapeOrDataForValue(
+    infer_context->SetShapeOrDataForValue(
         op->result(0),
         symbol::TensorShapeOrDataDimExprs{
             operand_shape_or_data.data().value()});