reminding message for TVM_REGISTER_NODE_TYPE (apache#4365)

zxy844288792 · Nov 26, 2019 · 01cdf99 · 01cdf99
1 parent f470272
commit 01cdf99
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/runtime/object.cc b/src/runtime/object.cc
@@ -155,7 +155,8 @@ class TypeContext {
   uint32_t TypeKey2Index(const std::string& skey) {
     auto it = type_key2index_.find(skey);
     CHECK(it != type_key2index_.end())
-        << "Cannot find type " << skey;
+        << "Cannot find type " << skey
+        << ". Did you forget to register the node by TVM_REGISTER_NODE_TYPE ?";
     return it->second;
   }