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