From 7ce8f66855362abda5227c92c1c82a21dc5e0959 Mon Sep 17 00:00:00 2001 From: Jared Roesch Date: Mon, 25 Jan 2021 15:16:11 -0800 Subject: [PATCH] Tweak printing for now --- src/relay/ir/function.cc | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/src/relay/ir/function.cc b/src/relay/ir/function.cc index c9920a621b56f..f6e87bd0fba71 100644 --- a/src/relay/ir/function.cc +++ b/src/relay/ir/function.cc @@ -62,9 +62,11 @@ TVM_REGISTER_GLOBAL("relay.ir.Function") TVM_STATIC_IR_FUNCTOR(ReprPrinter, vtable) .set_dispatch([](const ObjectRef& ref, ReprPrinter* p) { - auto* node = static_cast(ref.get()); - p->stream << "FunctionNode(" << node->params << ", " << node->ret_type << ", " << node->body - << ", " << node->type_params << ", " << node->attrs << ")"; + // auto* node = static_cast(ref.get()); + // p->stream << "FunctionNode(" << node->params << ", " << node->ret_type << ", " << node->body + // << ", " << node->type_params << ", " << node->attrs << ")"; + // we should standardize behavior + p->stream << PrettyPrint(ref); }); } // namespace relay