[Dataflow]: nullptr check (apache#5176)

trevor-m · Apr 16, 2020 · b285086 · b285086
1 parent b30d28d
commit b285086
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/te/schedule/schedule_dataflow_rewrite.cc b/src/te/schedule/schedule_dataflow_rewrite.cc
@@ -588,8 +588,8 @@ void InjectInline(ScheduleNode* sch) {
             if (!new_value.same_as(new_body[j][0])) {
               changed[j] = true;
               const tir::ReduceNode* r = new_value.as<tir::ReduceNode>();
-              CHECK_EQ(new_body[j].size(), r->source.size());
               CHECK(r != nullptr);
+              CHECK_EQ(new_body[j].size(), r->source.size());
               for (size_t k = 0; k < new_body[j].size(); ++k) {
                 auto n = make_object<tir::ReduceNode>(*r);
                 n->value_index = static_cast<int>(k);