Merge pull request PaddlePaddle#46 from tc20042008/xk-cinn-trivalop-fuse

move group_pattern.InferShardableAxes to group_pattern_util.InferShar…
feifei-111 · Mar 10, 2024 · c947ada · c947ada
2 parents c6bcf2d + 6bf5f0e
commit c947ada
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/paddle/cinn/frontend/group_pattern.h b/paddle/cinn/frontend/group_pattern.h
@@ -143,6 +143,4 @@ namespace cinn::frontend {
 using ErrorGroupPattern = api::ErrorPattern<frontend::FrontendPattern>;
 using GroupPattern = api::OpTopoPattern<frontend::FrontendPattern>;
 
-std::unordered_map<pir::Value, ShardableAxes> InferShardableAxes(const std::vector<const pir::Operation*>& ops);
-
 }
diff --git a/paddle/cinn/frontend/group_pattern_util.h b/paddle/cinn/frontend/group_pattern_util.h
@@ -6,5 +6,6 @@
 namespace cinn::frontend {
 
 GroupPattern GenerateGroupPatternFromFusionOp(const cinn::dialect::FusionOp&);
+std::unordered_map<pir::Value, ShardableAxes> InferShardableAxes(const std::vector<const pir::Operation*>& ops);
 
 }