reminding message for TVM_REGISTER_NODE_TYPE (#4365)

7efb72e6 · Yizhi Liu · GitHub · 3914be5e · 7efb72e6
Unverified Commit 7efb72e6 authored Nov 18, 2019 by Yizhi Liu Committed by GitHub Nov 18, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletions

src/runtime/object.cc
+2 -1

No files found.
--- 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;
  }