fix merge bug (PaddlePaddle#64069)

yinfan98 · May 7, 2024 · 39d494a · 39d494a
1 parent c4b9cc6
commit 39d494a
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()});