【Infer Symbolic Shape No.170】【BUAA】 Add read_file op (#67889)

* Finished read_file op * Resolved some suggested changes * Resolved some suggested changes * Removed unnecessary comment
PaddlePaddle · Sep 20, 2024 · 768e234 · 768e234
1 parent 319500d
commit 768e234
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 7 deletions.
diff --git a/paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/nullary_infer_sym.cc b/paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/nullary_infer_sym.cc
@@ -409,12 +409,24 @@ bool RandintOpInferSymbolicShape(
   }
 }
 
-// bool ReadFileOpInferSymbolicShape(pir::Operation *op,
-//                                   pir::InferSymbolicShapeContext
-//                                   *infer_context) {
-//   // pass
-//   return true;
-// }
+bool ReadFileOpInferSymbolicShape(
+    pir::Operation *op, pir::InferSymbolicShapeContext *infer_context) {
+  symbol::DimExpr unique_dim_sym = infer_context->GetNextSymName();
+
+  const std::vector<symbol::DimExpr> &out_shape = [&] {
+    std::vector<symbol::DimExpr> shape;
+    shape.emplace_back(symbol::DimExpr(1));
+    shape.emplace_back(unique_dim_sym);
+    return shape;
+  }();
+
+  infer_context->SetShapeOrDataForValue(
+      op->result(0),
+      symbol::ShapeOrDataDimExprs{
+          symbol::TensorShapeOrDataDimExprs(out_shape)});
+
+  return true;
+}
 
 // bool RecvV2OpInferSymbolicShape(pir::Operation *op,
 //                                 pir::InferSymbolicShapeContext

diff --git a/paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/nullary_infer_sym.h b/paddle/fluid/pir/dialect/operator/interface/infer_symbolic_shape/nullary_infer_sym.h
@@ -29,8 +29,8 @@ OP_DECLARE_INFER_SYMBOLIC_SHAPE(FullIntArray)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Gaussian)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Randint)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Randperm)
+OP_DECLARE_INFER_SYMBOLIC_SHAPE(ReadFile)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(Seed)
-// OP_DECLARE_INFER_SYMBOLIC_SHAPE(ReadFile)
 // OP_DECLARE_INFER_SYMBOLIC_SHAPE(RecvV2)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(TrilIndices)
 OP_DECLARE_INFER_SYMBOLIC_SHAPE(TriuIndices)

diff --git a/paddle/phi/ops/yaml/ops.yaml b/paddle/phi/ops/yaml/ops.yaml
@@ -3839,6 +3839,7 @@
     param : [filename]
     data_type : dtype
     backend : place
+  interfaces : paddle::dialect::InferSymbolicShapeInterface
 
 - op : real
   args : (Tensor x)