【Infer Symbolic Shape BUAA No.205】Add npu_identity op (#67500)

PaddlePaddle · Aug 21, 2024 · 7084558 · 7084558
1 parent fb57ee7
commit 7084558
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
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
@@ -195,6 +195,7 @@ OP_SAME_OPERANDS_AND_RESULT(GaussianInplace_)
 OP_SAME_OPERANDS_AND_RESULT(Hardshrink)
 OP_SAME_OPERANDS_AND_RESULT(HardSigmoid)
 OP_SAME_OPERANDS_AND_RESULT(MergeSelectedRows)
+OP_SAME_OPERANDS_AND_RESULT(NpuIdentity)
 OP_SAME_OPERANDS_AND_RESULT(Renorm)
 OP_SAME_OPERANDS_AND_RESULT(Renorm_)
 OP_SAME_OPERANDS_AND_RESULT(TanhShrink)

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
@@ -188,6 +188,7 @@ OP_DECLARE_INFER_SYMBOLIC_SHAPE(GaussianInplace_)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Hardshrink)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(HardSigmoid)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(MergeSelectedRows)
+OP_DECLARE_INFER_SYMBOLIC_SHAPE(NpuIdentity)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Renorm)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Renorm_)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(TanhShrink)

diff --git a/paddle/phi/ops/yaml/ops.yaml b/paddle/phi/ops/yaml/ops.yaml
@@ -3453,7 +3453,7 @@
     param : [x]
   kernel :
     func : npu_identity
-  # interfaces : paddle::dialect::InferSymbolicShapeInterface
+  interfaces : paddle::dialect::InferSymbolicShapeInterface
 
 - op : numel
   args : (Tensor x)