Signed-off-by: Wei Pan <weip@nvidia.com>
* \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
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;