diff --git a/include/tvm/node/repr_printer.h b/include/tvm/node/repr_printer.h index 41789a34d342..57824306620c 100644 --- 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 index e81a8241cb51..bf41c82f5a76 100644 --- 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(n)); +} + TVM_REGISTER_GLOBAL("node.AsRepr") .set_body_typed([](runtime::ObjectRef obj) { std::ostringstream os;