添加log2 log10符号推导 (#67851)

Lans1ot · web-flow · commit 6162f76da7a5 · 2024-09-03T10:20:03.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
@@ -112,6 +112,10 @@ OP_SAME_OPERANDS_AND_RESULT(Log1p)
 OP_SAME_OPERANDS_AND_RESULT(Log1p_)
 OP_SAME_OPERANDS_AND_RESULT(Log)
 OP_SAME_OPERANDS_AND_RESULT(Log_)
+OP_SAME_OPERANDS_AND_RESULT(Log10)
+OP_SAME_OPERANDS_AND_RESULT(Log10_)
+OP_SAME_OPERANDS_AND_RESULT(Log2)
+OP_SAME_OPERANDS_AND_RESULT(Log2_)
 OP_SAME_OPERANDS_AND_RESULT(LogicalNot)
 OP_SAME_OPERANDS_AND_RESULT(LogicalNot_)
 OP_SAME_OPERANDS_AND_RESULT(Logit)
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
@@ -103,6 +103,10 @@ OP_DECLARE_INFER_SYMBOLIC_SHAPE(Log1p)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Log1p_)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Log)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Log_)
+OP_DECLARE_INFER_SYMBOLIC_SHAPE(Log10)
+OP_DECLARE_INFER_SYMBOLIC_SHAPE(Log10_)
+OP_DECLARE_INFER_SYMBOLIC_SHAPE(Log2)
+OP_DECLARE_INFER_SYMBOLIC_SHAPE(Log2_)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(LogicalNot)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(LogicalNot_)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Logit)
diff --git a/paddle/phi/ops/yaml/ops.yaml b/paddle/phi/ops/yaml/ops.yaml
@@ -2850,6 +2850,7 @@
     func : log10
   inplace: (x -> out)
   backward: log10_grad
+  interfaces : paddle::dialect::InferSymbolicShapeInterface
 
 - op : log1p
   args : (Tensor x)
@@ -2871,6 +2872,7 @@
     func : log2
   inplace: (x -> out)
   backward: log2_grad
+  interfaces : paddle::dialect::InferSymbolicShapeInterface
 
 - op : log_loss
   args : (Tensor input, Tensor label, float epsilon)
