[IR] Fix a primitive check error (apache#5991)

* fix primitive check error * assuming every Op has Type defined * CHECK_NE -> CHECK Co-authored-by: Liangfu Chen <[email protected]>
neo-ai · Jul 14, 2020 · 6b18557 · 6b18557
1 parent e43b685
commit 6b18557
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/include/tvm/ir/op.h b/include/tvm/ir/op.h
@@ -146,6 +146,7 @@ class OpNode : public RelayExprNode {
   // Internal function to compute if it is primitive op
   bool IsPrimitiveOp_() const {
     const auto& fn_ty = this->op_type;
+    CHECK(fn_ty.get() != nullptr);
     if (fn_ty->type_constraints.size() != 1) return false;
     const TypeRelationNode* rel = fn_ty->type_constraints[0].as<TypeRelationNode>();
     if (rel == nullptr) return false;