From: Wei Pan <60017475+wpan11nv@users.noreply.github.com> Date: Thu, 2 Apr 2020 02:44:52 +0000 (-0700) Subject: [Debug] Add Dump function for Object type (NFC) (#5207) X-Git-Tag: upstream/0.7.0~1004 X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=15b1751c06b29862d5a665d89c6a11c829d5bcfe;p=platform%2Fupstream%2Ftvm.git [Debug] Add Dump function for Object type (NFC) (#5207) Signed-off-by: Wei Pan --- diff --git a/include/tvm/node/repr_printer.h b/include/tvm/node/repr_printer.h index 41789a3..5782430 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 e81a824..bf41c82 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;