Merge pull request PaddlePaddle#15 from Fridge003/multi-down

update LiftToAnchorPattern
Fridge003 · Apr 29, 2024 · eb6ef40 · eb6ef40
2 parents b0172b4 + 8e7d785
commit eb6ef40
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/paddle/cinn/operator_fusion/graph_transformer/operation.h b/paddle/cinn/operator_fusion/graph_transformer/operation.h
@@ -132,7 +132,9 @@ struct LiftToHorizontalFusionPatternOperation {
 struct LiftToAnchorPatternOperation {
   template <typename Phrase>
   void operator()(PatternGraph<Phrase>* graph, PatternNodePtr<Phrase> node) {
-    node->set_stmt_pattern(AnchorPattern<Phrase>(node->stmt_pattern()));
+    std::vector<pir::Operation*> ops = GetOpsInPattern(pattern);
+    pir::Value anchor = node->sink_op()->result(0);
+    node->set_stmt_pattern(AnchorPattern<Phrase>(ops, anchor));
   }
 };
 

diff --git a/paddle/cinn/operator_fusion/pattern.h b/paddle/cinn/operator_fusion/pattern.h
@@ -91,7 +91,7 @@ struct AnchorPattern {
       const std::vector<pir::Operation*>& ops,
       const pir::Value& anchor const AnchorState<T>& anchor_state)
       : ops_(ops), anchor_(anchor), {
-    // TODO(@wuzhanfei)
+    // TODO(@wuzhanfei): Initialize anchor state
   }
   std::vector<pir::Operation*> ops_;
   pir::Value anchor_;  // Choose only one anchor