[TE] Minor bugfix in message_passing.cc (apache#5254)

neo-ai · Apr 17, 2020 · 18acfd2 · 18acfd2
1 parent 559041e
commit 18acfd2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/te/schedule/message_passing.cc b/src/te/schedule/message_passing.cc
@@ -425,9 +425,9 @@ void PassUpBitMaskOr(const Stage& stage,
         continue;
       }
       int res = 0;
-      if (!state.count(s->parent)) res |= state[s->parent];
-      if (!state.count(s->inner)) res |= state[s->inner];
-      if (!state.count(s->outer)) res |= state[s->outer];
+      if (state.count(s->parent)) res |= state[s->parent];
+      if (state.count(s->inner)) res |= state[s->inner];
+      if (state.count(s->outer)) res |= state[s->outer];
       state[s->parent] = res;
     } else if (const FuseNode* s = rel.as<FuseNode>()) {
       if (!state.count(s->fused)) {