From b32acc328892612f2ed7845261fd2aff23ab5c4a Mon Sep 17 00:00:00 2001
From: Pratik Fegade <pratikfegade@gmail.com>
Date: Mon, 6 Apr 2020 18:28:11 -0400
Subject: [PATCH] [TE] Minor bugfix in message_passing.cc

---
 src/te/schedule/message_passing.cc | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/te/schedule/message_passing.cc b/src/te/schedule/message_passing.cc
index fc1327817020..4ff8586bccdf 100644
--- 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)) {