feat(fusion): reduce fusion reduce

PaddlePaddle · May 6, 2023 · b861b14 · b861b14
1 parent 9b41192
commit b861b14
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cinn/hlir/pass/fusion_merge_pass.cc b/cinn/hlir/pass/fusion_merge_pass.cc
@@ -838,7 +838,7 @@ class FusionMergePassHelper : public FusionHelperBase {
           } else {
             for (auto& belong_group : cur->belong_groups) {
               if (belong_group->group_id == cur->group_id) {
-                updated_consumers.insert(cur);
+                updated_consumers.insert(belong_group);
               } else {
                 fused_groups.push(belong_group);
               }