Fix loop error

jiahy0825 · Jun 13, 2023 · bc7d3ad · bc7d3ad
1 parent 5fa2c7d
commit bc7d3ad
Showing 1 changed file with 7 additions and 12 deletions.
diff --git a/cinn/hlir/pass/general_fusion_merge_pass.cc b/cinn/hlir/pass/general_fusion_merge_pass.cc
@@ -875,16 +875,12 @@ class FusionMergePassHelper : public FusionHelperBase {
       }
       return ret;
     };
-    bool update = false;
-    while (true) {
-      const auto& groups = GetFusableConsumerGroupList();
-      if (groups.size() <= 1) {
-        break;
-      }
-      HorizontalFuse(groups);
-      update = true;
+    const auto& groups = GetFusableConsumerGroupList();
+    if (groups.size() <= 1) {
+      return false;
     }
-    return update;
+    HorizontalFuse(groups);
+    return true;
   }
 
   bool HorizontalFusion(GroupPtr producer, const std::unordered_set<GroupPtr>& consumers) {
@@ -1675,12 +1671,11 @@ class FusionMergePassHelper : public FusionHelperBase {
         continue;
       }
       // do input fusion.
-      auto st = CallGeneralInputFusePass(input_consumers.second);
-      if (st) {
+      while (CallGeneralInputFusePass(input_consumers.second)) {
         // fused consumers, update
         UpdateInputToConsumers();
+        updated = true;
       }
-      updated |= st;
     }
 
     return updated;