[CINN][BUAA][Infer Symbolic Shape BUAA No.4,109]Add stanh and floor_divide op (#67788)

Marcusryz · web-flow · commit 01e753524a17 · 2024-09-02T10:11:35.000+08:00
* Add lerp CINN.

* update

* Refine lerp op CINN.

* Refine code.

* Refine code.

* Refine code.

* upstream

* add stanh and floordiv op cinn

* refine code
diff --git a/paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/element_wise_binary.cc b/paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/element_wise_binary.cc
@@ -152,6 +152,7 @@ OP_ELEMENT_WISE_BINARY(Divide_)
 OP_ELEMENT_WISE_BINARY(ElementwisePow)
 OP_ELEMENT_WISE_BINARY(Equal)
 OP_ELEMENT_WISE_BINARY(Equal_)
+OP_ELEMENT_WISE_BINARY(FloorDivide)
 OP_ELEMENT_WISE_BINARY(Fmax)
 OP_ELEMENT_WISE_BINARY(Fmin)
 OP_ELEMENT_WISE_BINARY(Gammaincc)
diff --git a/paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/element_wise_binary.h b/paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/element_wise_binary.h
@@ -37,6 +37,7 @@ OP_DECLARE_INFER_SYMBOLIC_SHAPE(Divide_)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(ElementwisePow)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Equal)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Equal_)
+OP_DECLARE_INFER_SYMBOLIC_SHAPE(FloorDivide)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Fmax)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Fmin)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Gammaincc)
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
@@ -161,6 +161,7 @@ OP_SAME_OPERANDS_AND_RESULT(Softplus)
 OP_SAME_OPERANDS_AND_RESULT(SoftRelu)
 OP_SAME_OPERANDS_AND_RESULT(Softshrink)
 OP_SAME_OPERANDS_AND_RESULT(Softsign)
+OP_SAME_OPERANDS_AND_RESULT(Stanh)
 OP_SAME_OPERANDS_AND_RESULT(Swish)
 OP_SAME_OPERANDS_AND_RESULT(Tan)
 OP_SAME_OPERANDS_AND_RESULT(Tan_)
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
@@ -153,6 +153,7 @@ OP_DECLARE_INFER_SYMBOLIC_SHAPE(Softplus)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(SoftRelu)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Softshrink)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Softsign)
+OP_DECLARE_INFER_SYMBOLIC_SHAPE(Stanh)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Swish)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Tan)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Tan_)
diff --git a/paddle/phi/ops/yaml/legacy/static_ops.yaml b/paddle/phi/ops/yaml/legacy/static_ops.yaml
@@ -328,6 +328,7 @@
   kernel :
     func : floor_divide
   traits : paddle::dialect::ForwardOnlyTrait
+  interfaces : paddle::dialect::InferSymbolicShapeInterface
 
 - op : frobenius_norm
   args : (Tensor x, IntArray axis={0}, bool keepdim=false, bool reduce_all=false, int in_dtype=-1, int out_dtype=-1)
diff --git a/paddle/phi/ops/yaml/ops.yaml b/paddle/phi/ops/yaml/ops.yaml
@@ -4555,6 +4555,7 @@
   kernel :
     func : stanh
   backward : stanh_grad
+  interfaces : paddle::dialect::InferSymbolicShapeInterface
 
 - op : stft
   args: (Tensor x, Tensor window, int n_fft, int hop_length, bool normalized, bool onesided)
