From 6784694d9a2e659d5b52c226d267b58df362e418 Mon Sep 17 00:00:00 2001
From: Andrew Liu <andrewlliu@gmail.com>
Date: Thu, 20 Aug 2020 21:42:09 -0700
Subject: [PATCH] fix broadcastcomprel

---
 src/relay/op/type_relations.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/relay/op/type_relations.cc b/src/relay/op/type_relations.cc
index dd9464a3f946e..169f7d388fecc 100644
--- 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;
 }