From 6bf5f0effb9f327924cf6eaf3f469bca7c7a3a00 Mon Sep 17 00:00:00 2001
From: jiahy0825 <jiahongyu@baidu.com>
Date: Sun, 10 Mar 2024 07:22:32 +0000
Subject: [PATCH] move group_pattern.InferShardableAxes to
 group_pattern_util.InferShardableAxes

---
 paddle/cinn/frontend/group_pattern.h      | 2 --
 paddle/cinn/frontend/group_pattern_util.h | 1 +
 2 files changed, 1 insertion(+), 2 deletions(-)

diff --git a/paddle/cinn/frontend/group_pattern.h b/paddle/cinn/frontend/group_pattern.h
index 9c9d7d4c638d8..ea69cc1db06ca 100644
--- 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);
-
 }
\ No newline at end of file
diff --git a/paddle/cinn/frontend/group_pattern_util.h b/paddle/cinn/frontend/group_pattern_util.h
index 9a2d919b3a4b9..da46b2be050af 100644
--- 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);
 
 }
\ No newline at end of file