From 318f894f04be9edd0afe4ae3267586b1e1334405 Mon Sep 17 00:00:00 2001
From: Eric Lunderberg <Lunderberg@users.noreply.github.com>
Date: Sat, 13 May 2023 19:03:34 -0500
Subject: [PATCH] [TVMScript] Print Executor, Runtime, and FunctionInfo as
 metadata (#14840)

Instances of `relay::Executor`, `relay::Runtime`, and
`relay::backend::FunctionInfo` may be inserted into a PrimFunc's
attributes for bookkeeping purposes.  In these flows, the function
should still be able to be printed for debugging purposes.
---
 src/script/printer/ir/relay.cc | 48 ++++++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)
 create mode 100644 src/script/printer/ir/relay.cc

diff --git a/src/script/printer/ir/relay.cc b/src/script/printer/ir/relay.cc
new file mode 100644
index 000000000000..574c07e32aa0
--- /dev/null
+++ b/src/script/printer/ir/relay.cc
@@ -0,0 +1,48 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+#include <tvm/relay/executor.h>
+#include <tvm/relay/runtime.h>
+
+#include "../../../relay/backend/utils.h"
+#include "./utils.h"
+
+namespace tvm {
+namespace script {
+namespace printer {
+
+TVM_STATIC_IR_FUNCTOR(IRDocsifier, vtable)
+    .set_dispatch<relay::Executor>("", [](relay::Executor ty, ObjectPath p, IRDocsifier d) -> Doc {
+      return d->AddMetadata(ty);
+    });
+
+TVM_STATIC_IR_FUNCTOR(IRDocsifier, vtable)
+    .set_dispatch<relay::Runtime>("", [](relay::Runtime ty, ObjectPath p, IRDocsifier d) -> Doc {
+      return d->AddMetadata(ty);
+    });
+
+TVM_STATIC_IR_FUNCTOR(IRDocsifier, vtable)
+    .set_dispatch<relay::backend::FunctionInfo>("",
+                                                [](relay::backend::FunctionInfo ty, ObjectPath p,
+                                                   IRDocsifier d) -> Doc {
+                                                  return d->AddMetadata(ty);
+                                                });
+
+}  // namespace printer
+}  // namespace script
+}  // namespace tvm