PaddlePaddle
diff --git a/‎cmake/cinn.cmake
Copy file name to clipboard
+12-6Lines changed: 12 additions & 6 deletions b/‎cmake/cinn.cmake
Copy file name to clipboard
+12-6Lines changed: 12 additions & 6 deletions
diff --git a/‎paddle/cinn/frontend/CMakeLists.txt
Copy file name to clipboard
+1-1Lines changed: 1 addition & 1 deletion b/‎paddle/cinn/frontend/CMakeLists.txt
Copy file name to clipboard
+1-1Lines changed: 1 addition & 1 deletion
diff --git a/‎paddle/cinn/frontend/group_cluster/CMakeLists.txt
Copy file name to clipboard
+4-1Lines changed: 4 additions & 1 deletion b/‎paddle/cinn/frontend/group_cluster/CMakeLists.txt
Copy file name to clipboard
+4-1Lines changed: 4 additions & 1 deletion
diff --git a/‎paddle/cinn/frontend/group_cluster/cluster_policy/CMakeLists.txt
Copy file name to clipboard
+2-2Lines changed: 2 additions & 2 deletions b/‎paddle/cinn/frontend/group_cluster/cluster_policy/CMakeLists.txt
Copy file name to clipboard
+2-2Lines changed: 2 additions & 2 deletions
diff --git a/‎paddle/cinn/frontend/group_cluster/cluster_policy/general_topo_policy.cc
Copy file name to clipboard
+21-3Lines changed: 21 additions & 3 deletions b/‎paddle/cinn/frontend/group_cluster/cluster_policy/general_topo_policy.cc
Copy file name to clipboard
+21-3Lines changed: 21 additions & 3 deletions
diff --git a/‎paddle/cinn/frontend/group_cluster/cluster_policy/general_topo_policy.h
Copy file name to clipboard
+3-1Lines changed: 3 additions & 1 deletion b/‎paddle/cinn/frontend/group_cluster/cluster_policy/general_topo_policy.h
Copy file name to clipboard
+3-1Lines changed: 3 additions & 1 deletion
diff --git a/‎paddle/cinn/frontend/group_cluster/cluster_policy/policy_manager.cc
Copy file name to clipboard
+12-2Lines changed: 12 additions & 2 deletions b/‎paddle/cinn/frontend/group_cluster/cluster_policy/policy_manager.cc
Copy file name to clipboard
+12-2Lines changed: 12 additions & 2 deletions
diff --git a/‎paddle/cinn/frontend/group_cluster/cluster_policy/policy_manager.h
Copy file name to clipboard
+11-3Lines changed: 11 additions & 3 deletions b/‎paddle/cinn/frontend/group_cluster/cluster_policy/policy_manager.h
Copy file name to clipboard
+11-3Lines changed: 11 additions & 3 deletions
@@ -164,6 +164,8 @@ cinn_cc_library(
   isl
   ginac
   pybind
+  group_cluster
+  cinn_op_dialect
   ${jitify_deps})
 add_dependencies(cinnapi GEN_LLVM_RUNTIME_IR_HEADER ZLIB::ZLIB)
 add_dependencies(cinnapi GEN_LLVM_RUNTIME_IR_HEADER ${core_deps})
@@ -220,13 +222,17 @@ function(gen_cinncore LINKTYPE)
     schedule_desc_proto
     absl
     isl
-    ginac)
+    ginac
+    pybind
+    group_cluster
+    cinn_op_dialect
+    ${jitify_deps})
   add_dependencies(${CINNCORE_TARGET} GEN_LLVM_RUNTIME_IR_HEADER ZLIB::ZLIB)
   add_dependencies(${CINNCORE_TARGET} GEN_LLVM_RUNTIME_IR_HEADER ${core_deps})
   target_link_libraries(${CINNCORE_TARGET} op_dialect pir phi)
   add_dependencies(${CINNCORE_TARGET} op_dialect pir phi)
 
-  add_dependencies(${CINNCORE_TARGET} pybind)
+  # add_dependencies(${CINNCORE_TARGET} pybind)
   target_link_libraries(${CINNCORE_TARGET} ${PYTHON_LIBRARIES})
 
   if(WITH_MKL)
@@ -247,16 +253,16 @@ function(gen_cinncore LINKTYPE)
       ${CUBLAS}
       ${CUDNN}
       ${CURAND}
-      ${CUSOLVER}
-      ${jitify_deps})
+      ${CUSOLVER})
+    # ${jitify_deps})
     if(NVTX_FOUND)
       target_link_libraries(${CINNCORE_TARGET} ${CUDA_NVTX_LIB})
     endif()
   endif()
 
   if(WITH_CUTLASS)
-    target_link_libraries(cinnapi cutlass)
-    add_dependencies(cinnapi cutlass)
+    target_link_libraries(${CINNCORE_TARGET} cutlass)
+    add_dependencies(${CINNCORE_TARGET} cutlass)
   endif()
 endfunction()
 
 
@@ -62,7 +62,7 @@ add_subdirectory(paddle)
 add_subdirectory(decomposer)
 add_subdirectory(op_mappers)
 add_subdirectory(pass)
-# add_subdirectory(group_cluster)
+add_subdirectory(group_cluster)
 
 cinn_cc_test(test_op_mapper_registry SRCS op_mapper_registry_test.cc DEPS
              cinncore)
@@ -3,4 +3,7 @@ gather_srcs(group_cluster_src SRCS common_utils.cc pattern_node.cc
 
 add_subdirectory(cluster_policy)
 
-cc_library(group_cluster SRCS ${group_cluster_src})
+cc_library(
+  group_cluster
+  SRCS ${group_cluster_src}
+  DEPS phi)
@@ -1,3 +1,3 @@
-gather_srcs(group_cluster_src SRCS general_topo_policy.cc policy_manager.cc)
-
+gather_srcs(group_cluster_src SRCS general_topo_policy.cc policy_manager.cc
+            relative_judge_policy.cc)
 add_subdirectory(shardable_axes_policy)
@@ -16,10 +16,28 @@
 
 namespace cinn::frontend::group_cluster::policy {
 
-bool GeneralTopoPolicy::CanFuse(const PatternNodePtr upstream,
-                                const PatternNodePtr downstream) {
-  // TODO(wuzhanfei) topo policy (if lead to loop)
+bool IsDownstreamNode(const PatternNodePtr start, const PatternNodePtr target) {
+  if (start == target) return true;
+  for (const auto& down_node : start->downstream_) {
+    if (IsDownstreamNode(down_node, target)) return true;
+  }
   return false;
 }
 
+bool IsIndirectDownstreamNode(const PatternNodePtr start,
+                              const PatternNodePtr target) {
+  for (const auto& node : start->downstream_) {
+    if (node == target) continue;
+    if (IsDownstreamNode(node, target)) return true;
+  }
+  return false;
+}
+
+bool GeneralTopoPolicy::CanFuse(const PatternNodePtr& first,
+                                const PatternNodePtr& second) {
+  VLOG(4) << "Start GeneralTopoPolicy";
+  return !(IsIndirectDownstreamNode(first, second) ||
+           IsIndirectDownstreamNode(second, first));
+}
+
 }  // namespace cinn::frontend::group_cluster::policy
@@ -19,7 +19,9 @@ namespace cinn::frontend::group_cluster::policy {
 
 class GeneralTopoPolicy final : virtual public Policy {
  public:
-  bool CanFuse(const PatternNodePtr upstream, const PatternNodePtr downstream);
+  bool CanFuse(const PatternNodePtr& upstream,
+               const PatternNodePtr& downstream) override;
+  std::string Name() { return "GeneralTopoPolicy"; }
 };
 
 }  // namespace cinn::frontend::group_cluster::policy
@@ -17,12 +17,22 @@
 
 namespace cinn::frontend::group_cluster::policy {
 
-bool PolicyManager::CanFuse(const PatternNodePtr upstream,
-                            const PatternNodePtr downstream) {
+bool PolicyManager::CanFuse(const PatternNodePtr& upstream,
+                            const PatternNodePtr& downstream) const {
   for (const auto& policy : policies_) {
     if (!policy->CanFuse(upstream, downstream)) return false;
   }
   return true;
 }
 
+std::vector<size_t> PolicyManager::GetFakeReduceIterIdx(
+    const PatternNodePtr& upstream, const PatternNodePtr& downstream) const {
+  for (const auto& policy : policies_) {
+    if (policy->Name() == "RelativeJudgePolicy") {
+      return policy->GetFakeReduceIterIdx(upstream, downstream);
+    }
+  }
+  return {};
+}
+
 }  // namespace cinn::frontend::group_cluster::policy
@@ -20,8 +20,13 @@ namespace cinn::frontend::group_cluster::policy {
 
 class Policy {
  public:
-  virtual bool CanFuse(const PatternNodePtr upstream,
-                       const PatternNodePtr downstream) = 0;
+  virtual std::string Name() = 0;
+  virtual bool CanFuse(const PatternNodePtr& upstream,
+                       const PatternNodePtr& downstream) = 0;
+  virtual std::vector<size_t> GetFakeReduceIterIdx(
+      const PatternNodePtr& upstream, const PatternNodePtr& downstream) {
+    return {};
+  }
 };
 
 using PolicyPtr = std::shared_ptr<Policy>;
@@ -30,7 +35,10 @@ class PolicyManager {
  public:
   explicit PolicyManager(const std::vector<PolicyPtr>& policies)
       : policies_(policies) {}
-  bool CanFuse(const PatternNodePtr upstream, const PatternNodePtr downstream);
+  bool CanFuse(const PatternNodePtr& upstream,
+               const PatternNodePtr& downstream) const;
+  std::vector<size_t> GetFakeReduceIterIdx(
+      const PatternNodePtr& upstream, const PatternNodePtr& downstream) const;
 
  private:
   std::vector<PolicyPtr> policies_;