From 15b1751c06b29862d5a665d89c6a11c829d5bcfe Mon Sep 17 00:00:00 2001 From: Wei Pan <60017475+wpan11nv@users.noreply.github.com> Date: Wed, 1 Apr 2020 19:44:52 -0700 Subject: [PATCH] [Debug] Add Dump function for Object type (NFC) (#5207) Signed-off-by: Wei Pan --- include/tvm/node/repr_printer.h | 8 +++++++- src/node/repr_printer.cc | 6 +++++- 2 files changed, 12 insertions(+), 2 deletions(-) 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;