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 index bee1038bb65ef..fbd84f5cb4d53 100644 --- 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 index 52851c7b26411..6c08fd1e35d70 100644 --- 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 index a607870f3f475..63bc0d188d253 100755 --- 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