[Debug] Add Dump function for Object type (NFC) (#5207)

Signed-off-by: Wei Pan <weip@nvidia.com>
apache · Apr 2, 2020 · 15b1751 · 15b1751
1 parent 5b857d3
commit 15b1751
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/include/tvm/node/repr_printer.h b/include/tvm/node/repr_printer.h
@@ -51,7 +51,13 @@ class ReprPrinter {
  * \brief Dump the node to stderr, used for debug purposes.
  * \param node The input node
  */
-TVM_DLL void Dump(const ObjectRef& node);
+TVM_DLL void Dump(const runtime::ObjectRef& node);
+
+/*!
+ * \brief Dump the node to stderr, used for debug purposes.
+ * \param node The input node
+ */
+TVM_DLL void Dump(const runtime::Object* node);
 
 }  // namespace tvm
 

diff --git a/src/node/repr_printer.cc b/src/node/repr_printer.cc
@@ -51,10 +51,14 @@ ReprPrinter::FType& ReprPrinter::vtable() {
   return inst;
 }
 
-void Dump(const ObjectRef& n) {
+void Dump(const runtime::ObjectRef& n) {
   std::cerr << n << "\n";
 }
 
+void Dump(const runtime::Object* n) {
+  Dump(runtime::GetRef<runtime::ObjectRef>(n));
+}
+
 TVM_REGISTER_GLOBAL("node.AsRepr")
 .set_body_typed([](runtime::ObjectRef obj) {
   std::ostringstream os;