update (PaddlePaddle#64042)

yinfan98 · May 7, 2024 · e37e36d · e37e36d
1 parent 3c186a0
commit e37e36d
Showing 1 changed file with 14 additions and 3 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
@@ -292,9 +292,20 @@ bool GaussianOpInferSymbolicShape(
     return true;
 
   } else {
-    PADDLE_THROW(phi::errors::Unimplemented(
-        "Currently shape must comes from FullIntArrayOp in GaussianOp's "
-        "InferSymbolicShape."));
+    pir::Value operand_source = op->operand_source(0);
+    const symbol::ShapeOrDataDimExprs &operand_shape_or_data =
+        shape_analysis->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(
+        op->result(0),
+        symbol::TensorShapeOrDataDimExprs{
+            operand_shape_or_data.data().value()});
     return true;
   }
 }