【Infer Symbolic Shape [BUAA] No.93 & 98 & 105 】sqrt & silu & view_dtype (#67008)

BHmingyang · web-flow · commit 66f17e7faade · 2024-08-07T16:08:41.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
@@ -176,6 +176,12 @@ OP_SAME_OPERANDS_AND_RESULT(Polygamma)
 OP_SAME_OPERANDS_AND_RESULT(Polygamma_)
 OP_SAME_OPERANDS_AND_RESULT(EnableCheckModelNanInf)
 OP_SAME_OPERANDS_AND_RESULT(ViewShape)
+OP_SAME_OPERANDS_AND_RESULT(Silu)
+OP_SAME_OPERANDS_AND_RESULT(ViewDtype)
+OP_SAME_OPERANDS_AND_RESULT(Sqrt)
+OP_SAME_OPERANDS_AND_RESULT(Sqrt_)
+OP_SAME_OPERANDS_AND_RESULT(SqrtSr)
+OP_SAME_OPERANDS_AND_RESULT(SqrtSr_)
 OP_SAME_OPERANDS_AND_RESULT(FusedSoftmaxMaskUpperTriangle)
 OP_SAME_OPERANDS_AND_RESULT(Gammaln)
 OP_SAME_OPERANDS_AND_RESULT(Gammaln_)
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
@@ -137,6 +137,7 @@ OP_DECLARE_INFER_SYMBOLIC_SHAPE(Select)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(ShadowFeed)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(ShareData_)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Sign)
+OP_DECLARE_INFER_SYMBOLIC_SHAPE(Silu)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Sin)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Sin_)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Sinh)
@@ -169,6 +170,11 @@ OP_DECLARE_INFER_SYMBOLIC_SHAPE(Polygamma)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Polygamma_)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(EnableCheckModelNanInf)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(ViewShape)
+OP_DECLARE_INFER_SYMBOLIC_SHAPE(ViewDtype)
+OP_DECLARE_INFER_SYMBOLIC_SHAPE(Sqrt)
+OP_DECLARE_INFER_SYMBOLIC_SHAPE(Sqrt_)
+OP_DECLARE_INFER_SYMBOLIC_SHAPE(SqrtSr)
+OP_DECLARE_INFER_SYMBOLIC_SHAPE(SqrtSr_)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(FusedSoftmaxMaskUpperTriangle)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Gammaln)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Gammaln_)
diff --git a/paddle/phi/ops/yaml/ops.yaml b/paddle/phi/ops/yaml/ops.yaml
@@ -4211,7 +4211,7 @@
   kernel :
     func : silu
   backward : silu_grad
-  interfaces : paddle::dialect::LayoutTransformationInterface
+  interfaces : paddle::dialect::LayoutTransformationInterface, paddle::dialect::InferSymbolicShapeInterface
 
 - op : sin
   args : (Tensor x)
@@ -4354,6 +4354,7 @@
            sqrt_sr {selected_rows -> selected_rows}
   inplace : (x -> out)
   backward : sqrt_grad
+  interfaces : paddle::dialect::InferSymbolicShapeInterface
 
 - op : square
   args : (Tensor x)
@@ -4881,6 +4882,7 @@
     data_type : input
   backward : view_dtype_grad
   no_need_buffer : input
+  interfaces : paddle::dialect::InferSymbolicShapeInterface
 
 - op : view_shape
   args : (Tensor input, int64_t[] dims = {})
