fix broadcastcomprel

apache · Aug 21, 2020 · 6784694 · 6784694
1 parent 7a2dc43
commit 6784694
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/relay/op/type_relations.cc b/src/relay/op/type_relations.cc
@@ -118,7 +118,6 @@ bool BroadcastCompRel(const Array<Type>& types, int num_inputs, const Attrs& att
   CHECK_EQ(types.size(), 3);
   // DLOG(INFO) << "In1:" << types[0] << ",In2:" << types[1]
   //                 << ",Out:" << types[2] << std::endl;
-  reporter->Assign(types[0], types[1]);
   if (auto* t0 = types[0].as<TensorTypeNode>()) {
     if (auto* t1 = types[1].as<TensorTypeNode>()) {
       CHECK_EQ(t0->dtype, t1->dtype);
@@ -127,6 +126,7 @@ bool BroadcastCompRel(const Array<Type>& types, int num_inputs, const Attrs& att
       return true;
     }
   }
+  reporter->Assign(types[0], types[1]);
   return false;
 }