fix broadcastcomprel

hypercubestart · hypercubestart · commit 6784694d9a2e · 2020-08-20T21:42:09.000-07:00
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;
 }
 

Original file line number	Diff line number	Diff line change
`@@ -118,7 +118,6 @@ bool BroadcastCompRel(const Array<Type>& types, int num_inputs, const Attrs& att`
`118`	`118`	`CHECK_EQ(types.size(), 3);`
`119`	`119`	`// DLOG(INFO) << "In1:" << types[0] << ",In2:" << types[1]`
`120`	`120`	`// << ",Out:" << types[2] << std::endl;`
`121`		`- reporter->Assign(types[0], types[1]);`
`122`	`121`	`if (auto* t0 = types[0].as<TensorTypeNode>()) {`
`123`	`122`	`if (auto* t1 = types[1].as<TensorTypeNode>()) {`
`124`	`123`	`CHECK_EQ(t0->dtype, t1->dtype);`
`@@ -127,6 +126,7 @@ bool BroadcastCompRel(const Array<Type>& types, int num_inputs, const Attrs& att`
`127`	`126`	`return true;`
`128`	`127`	`}`
`129`	`128`	`}`
	`129`	`+ reporter->Assign(types[0], types[1]);`
`130`	`130`	`return false;`
`131`	`131`	`}`
`132`	`132`