[TIR][Printer] text format printer considering future parsing use (#5483

)
apache · May 9, 2020 · ae7e0a1 · ae7e0a1
1 parent 06efa7f
commit ae7e0a1
Show file tree

Hide file tree

Showing 9 changed files with 1,765 additions and 739 deletions.
diff --git a/include/tvm/ir/module.h b/include/tvm/ir/module.h
@@ -363,7 +363,7 @@ TVM_DLL String PrettyPrint(const ObjectRef& node);
  * \return The text representation.
  */
 TVM_DLL String AsText(const ObjectRef& node,
-                           bool show_meta_data = true,
-                           runtime::TypedPackedFunc<String(ObjectRef)> annotate = nullptr);
+                      bool show_meta_data = true,
+                      runtime::TypedPackedFunc<String(ObjectRef)> annotate = nullptr);
 }  // namespace tvm
 #endif  // TVM_IR_MODULE_H_
diff --git a/src/printer/meta_data.h b/src/printer/meta_data.h
@@ -108,6 +108,15 @@ class TextMetaDataContext {
     return meta_repr_[node];
   }
 
+  /*!
+   * \brief Test whether a node has been put in meta
+   * \param node The query node
+   * \return whether the node has been put in meta
+   */
+  bool InMeta(const ObjectRef& node) {
+    return meta_repr_.find(node) != meta_repr_.end();
+  }
+
   /*!
    * \brief Print a key value pair
    */