【Infer Symbolic Shape BUAA No.51 BUAA】gelu (#66961)

Wizard-ZP · web-flow · commit cb2000306637 · 2024-08-06T00:34:35.000+08:00
diff --git a/paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/same_operands_result.cc b/paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/same_operands_result.cc
@@ -79,6 +79,8 @@ OP_SAME_OPERANDS_AND_RESULT(Flip)
 OP_SAME_OPERANDS_AND_RESULT(Floor)
 OP_SAME_OPERANDS_AND_RESULT(Floor_)
 OP_SAME_OPERANDS_AND_RESULT(FullLike)
+OP_SAME_OPERANDS_AND_RESULT(Gelu)
+OP_SAME_OPERANDS_AND_RESULT(Gelu_)
 OP_SAME_OPERANDS_AND_RESULT(Imag)
 OP_SAME_OPERANDS_AND_RESULT(Increment)
 OP_SAME_OPERANDS_AND_RESULT(Increment_)
diff --git a/paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/same_operands_result.h b/paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/same_operands_result.h
@@ -71,6 +71,8 @@ OP_DECLARE_INFER_SYMBOLIC_SHAPE(Flip)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Floor)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Floor_)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(FullLike)
+OP_DECLARE_INFER_SYMBOLIC_SHAPE(Gelu)
+OP_DECLARE_INFER_SYMBOLIC_SHAPE(Gelu_)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Imag)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Increment)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Increment_)
diff --git a/paddle/phi/ops/yaml/ops.yaml b/paddle/phi/ops/yaml/ops.yaml
@@ -2202,6 +2202,7 @@
   kernel :
     func : gelu
   backward : gelu_grad
+  interfaces : paddle::dialect::InferSymbolicShapeInterface
 
 - op : generate_proposals
   args : (Tensor scores, Tensor bbox_deltas, Tensor im_shape, Tensor anchors, Tensor variances, int pre_nms_top_n, int post_nms_top_n, float nms_thresh, float min_size, float eta, bool pixel_offset=true)
