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 index 1619df4a543a7..0fa04ee0d6eaa 100644 --- 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 @@ -144,6 +144,7 @@ OP_SAME_OPERANDS_AND_RESULT(ScatterNdAdd) OP_SAME_OPERANDS_AND_RESULT(Scatter) OP_SAME_OPERANDS_AND_RESULT(Scatter_) OP_SAME_OPERANDS_AND_RESULT(Select) +OP_SAME_OPERANDS_AND_RESULT(Selu) OP_SAME_OPERANDS_AND_RESULT(ShadowFeed) OP_SAME_OPERANDS_AND_RESULT(ShareData_) OP_SAME_OPERANDS_AND_RESULT(Sign) 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 index 9e67899f78bc8..a34c20a712e7c 100644 --- 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 @@ -167,6 +167,7 @@ OP_DECLARE_INFER_SYMBOLIC_SHAPE(LeakyRelu) OP_DECLARE_INFER_SYMBOLIC_SHAPE(LeakyRelu_) OP_DECLARE_INFER_SYMBOLIC_SHAPE(ThresholdedRelu) OP_DECLARE_INFER_SYMBOLIC_SHAPE(ThresholdedRelu_) +OP_DECLARE_INFER_SYMBOLIC_SHAPE(Selu) OP_DECLARE_INFER_SYMBOLIC_SHAPE(SquareSr) OP_DECLARE_INFER_SYMBOLIC_SHAPE(Square) OP_DECLARE_INFER_SYMBOLIC_SHAPE(Polygamma) diff --git a/paddle/phi/ops/yaml/ops.yaml b/paddle/phi/ops/yaml/ops.yaml index 0a460978d63f3..29e16f2209e65 100755 --- a/paddle/phi/ops/yaml/ops.yaml +++ b/paddle/phi/ops/yaml/ops.yaml @@ -4121,7 +4121,7 @@ kernel : func : selu backward : selu_grad - # interfaces : paddle::dialect::InferSymbolicShapeInterface + interfaces : paddle::dialect::InferSymbolicShapeInterface - op : send_u_recv args : (Tensor x, Tensor src_index, Tensor dst_index, str reduce_op = "SUM", IntArray out_size = {0})