From 00a848134c5d71edad45f3320a904cdf6fa8cab4 Mon Sep 17 00:00:00 2001 From: Pratik Fegade Date: Mon, 6 Apr 2020 22:04:20 -0400 Subject: [PATCH] [TE] Minor bugfix in message_passing.cc (#5254) --- 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 fc13278..4ff8586 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()) { if (!state.count(s->fused)) { -- 2.7.4