【Infer Symbolic Shape No.14】[BUAA] Add check_numerics op (#67735)

* Finished check_numerics op

* Fixed errors

* Fixed errors

Author: MufanColin

paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/unary_infer_sym.cc

@@ -496,6 +496,21 @@ bool Cast_OpInferSymbolicShape(pir::Operation *op,
   return CastOpInferSymbolicShape(op, infer_context);
 }
 
+bool CheckNumericsOpInferSymbolicShape(
+    pir::Operation *op, pir::InferSymbolicShapeContext *infer_context) {
+  infer_context->SetShapeOrDataForValue(
+      op->result(0),
+      symbol::ShapeOrDataDimExprs{
+          symbol::TensorShapeOrDataDimExprs({symbol::DimExpr(3)})});
+
+  infer_context->SetShapeOrDataForValue(
+      op->result(1),
+      symbol::ShapeOrDataDimExprs{
+          symbol::TensorShapeOrDataDimExprs({symbol::DimExpr(3)})});
+
+  return true;
+}
+
 bool CholeskyOpInferSymbolicShape(
     pir::Operation *op, pir::InferSymbolicShapeContext *infer_context) {
   const auto &x_shape =

paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/unary_infer_sym.h

@@ -45,6 +45,7 @@ OP_DECLARE_INFER_SYMBOLIC_SHAPE(Cumprod_)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Cumsum)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Cumsum_)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(ChannelShuffle)
+OP_DECLARE_INFER_SYMBOLIC_SHAPE(CheckNumerics)
 // OP_DECLARE_INFER_SYMBOLIC_SHAPE(Crop)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(DecodeJpeg)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Diag)

paddle/phi/ops/yaml/ops.yaml

@@ -875,6 +875,7 @@
     func : CheckNumericsInferMeta
   kernel :
     func : check_numerics
+  interfaces : paddle::dialect::InferSymbolicShapeInterface
 
 - op : cholesky
   args : (Tensor x, bool upper=false)