【Infer Symbolic Shape BUAA No.20 No.24 No.32 No.68 No.77】Add mish label_smooth elu disable_check_model_nan_inf depend op (#66889)

crazyxiaoxi · web-flow · commit 919b3ee19a90 · 2024-08-06T16:43:33.000+08:00
* resolve conflict

* fix
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
@@ -62,6 +62,10 @@ OP_SAME_OPERANDS_AND_RESULT(DequantizeLog)
 OP_SAME_OPERANDS_AND_RESULT(Digamma)
 OP_SAME_OPERANDS_AND_RESULT(Digamma_)
 OP_SAME_OPERANDS_AND_RESULT(Dirichlet)
+OP_SAME_OPERANDS_AND_RESULT(DisableCheckModelNanInf)
+OP_SAME_OPERANDS_AND_RESULT(Depend)
+OP_SAME_OPERANDS_AND_RESULT(Elu)
+OP_SAME_OPERANDS_AND_RESULT(Elu_)
 OP_SAME_OPERANDS_AND_RESULT(EmptyLike)
 OP_SAME_OPERANDS_AND_RESULT(Erf)
 OP_SAME_OPERANDS_AND_RESULT(Erf_)
@@ -95,6 +99,7 @@ OP_SAME_OPERANDS_AND_RESULT(I0_)
 OP_SAME_OPERANDS_AND_RESULT(I0e)
 OP_SAME_OPERANDS_AND_RESULT(I1)
 OP_SAME_OPERANDS_AND_RESULT(I1e)
+OP_SAME_OPERANDS_AND_RESULT(LabelSmooth)
 OP_SAME_OPERANDS_AND_RESULT(Lgamma)
 OP_SAME_OPERANDS_AND_RESULT(Lgamma_)
 OP_SAME_OPERANDS_AND_RESULT(Log1p)
@@ -107,6 +112,7 @@ OP_SAME_OPERANDS_AND_RESULT(Logit)
 OP_SAME_OPERANDS_AND_RESULT(Logit_)
 OP_SAME_OPERANDS_AND_RESULT(Logsigmoid)
 OP_SAME_OPERANDS_AND_RESULT(Logsigmoid_)
+OP_SAME_OPERANDS_AND_RESULT(Mish)
 OP_SAME_OPERANDS_AND_RESULT(Pow)
 OP_SAME_OPERANDS_AND_RESULT(Poisson)
 OP_SAME_OPERANDS_AND_RESULT(Pow_)
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
@@ -54,6 +54,10 @@ OP_DECLARE_INFER_SYMBOLIC_SHAPE(DequantizeLog)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Digamma)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Digamma_)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Dirichlet)
+OP_DECLARE_INFER_SYMBOLIC_SHAPE(DisableCheckModelNanInf)
+OP_DECLARE_INFER_SYMBOLIC_SHAPE(Depend)
+OP_DECLARE_INFER_SYMBOLIC_SHAPE(Elu)
+OP_DECLARE_INFER_SYMBOLIC_SHAPE(Elu_)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(EmptyLike)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Erf)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Erf_)
@@ -87,6 +91,7 @@ OP_DECLARE_INFER_SYMBOLIC_SHAPE(I0_)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(I0e)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(I1)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(I1e)
+OP_DECLARE_INFER_SYMBOLIC_SHAPE(LabelSmooth)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Lgamma)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Lgamma_)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Log1p)
@@ -99,6 +104,7 @@ OP_DECLARE_INFER_SYMBOLIC_SHAPE(Logit)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Logit_)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Logsigmoid)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Logsigmoid_)
+OP_DECLARE_INFER_SYMBOLIC_SHAPE(Mish)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Poisson)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Pow)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Pow_)
diff --git a/paddle/phi/ops/yaml/ops.yaml b/paddle/phi/ops/yaml/ops.yaml
@@ -1250,6 +1250,7 @@
     param : [x]
   kernel:
     func: depend
+  interfaces : paddle::dialect::InferSymbolicShapeInterface
 
 - op : depthwise_conv2d
   args : (Tensor input, Tensor filter, int[] strides={1, 1}, int[] paddings={0, 0}, str padding_algorithm="EXPLICIT", int groups=1, int[] dilations={1, 1}, str data_format="NCHW")
@@ -1414,6 +1415,7 @@
     func: check_model_nan_inf
     data_type: x
   backward: disable_check_model_nan_inf_grad
+  interfaces : paddle::dialect::InferSymbolicShapeInterface
 
 - op : dist
   args : (Tensor x, Tensor y, float p = 2.0)
@@ -1514,6 +1516,7 @@
     func : elu
   inplace : (x -> out)
   backward : elu_grad
+  interfaces : paddle::dialect::InferSymbolicShapeInterface
 
 - op : empty
   args : (IntArray shape, DataType dtype=DataType::FLOAT32, Place place=CPUPlace())
@@ -2649,6 +2652,7 @@
     data_type : label
   optional : prior_dist
   backward : label_smooth_grad
+  interfaces : paddle::dialect::InferSymbolicShapeInterface
 
 - op : lamb_
   args : (Tensor param, Tensor grad, Tensor learning_rate, Tensor moment1, Tensor moment2, Tensor beta1_pow, Tensor beta2_pow, Tensor master_param, Tensor skip_update, float weight_decay, float beta1=0.9, float beta2=0.999, float epsilon=1.0e-6f, bool always_adapt=false, bool multi_precision=false)
@@ -3209,6 +3213,7 @@
   kernel :
     func : mish
   backward : mish_grad
+  interfaces : paddle::dialect::InferSymbolicShapeInterface
 
 - op : mode
   args : (Tensor x,  int axis = -1,  bool keepdim = false)
