【Infer Symbolic Shape No.105】Add full_ implementation (#68236)

enkilee · web-flow · commit 248b2de03bce · 2024-10-16T20:56:23.000+08:00
* fix

* CI
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
@@ -309,6 +309,11 @@ bool FullOpInferSymbolicShape(pir::Operation *op,
   return true;
 }
 
+bool Full_OpInferSymbolicShape(pir::Operation *op,
+                               pir::InferSymbolicShapeContext *infer_context) {
+  return FullOpInferSymbolicShape(op, infer_context);
+}
+
 bool FullIntArrayOpInferSymbolicShape(
     pir::Operation *op, pir::InferSymbolicShapeContext *infer_context) {
   const auto &attributes = op->attributes();
diff --git a/paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/nullary_infer_sym.h b/paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/nullary_infer_sym.h
@@ -25,6 +25,7 @@ OP_DECLARE_INFER_SYMBOLIC_SHAPE(Data)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Empty)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Feed)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Full)
+OP_DECLARE_INFER_SYMBOLIC_SHAPE(Full_)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(FullIntArray)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Gaussian)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Randint)
diff --git a/paddle/phi/ops/yaml/ops.yaml b/paddle/phi/ops/yaml/ops.yaml
@@ -2141,6 +2141,7 @@
     param : [shape, value, dtype]
     data_type : dtype
     backend : place
+  interfaces : paddle::dialect::InferSymbolicShapeInterface
   traits : paddle::dialect::ForwardOnlyTrait
 
 - op : full_batch_size_like
