Add debug print for Visual Studio immediate window.

Now you can run tvm::ToString(node) to show the content
prateek9623 · Jun 28, 2019 · 69cbb52 · 69cbb52
1 parent fd48016
commit 69cbb52
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/include/tvm/expr.h b/include/tvm/expr.h
@@ -278,6 +278,12 @@ TVM_DLL std::ostream& operator<<(std::ostream& os, const NodeRef& n);  // NOLINT
  */
 TVM_DLL void Dump(const NodeRef& node);
 
+/*!
+ * \brief Dump the node to char[], used for debug purposes.
+ * \param node The input node
+ */
+TVM_DLL std::unique_ptr<char[]> ToString(const NodeRef& node);
+
 // definition of Node.
 /*!
  * \brief An iteration variable representing an iteration

diff --git a/src/lang/expr.cc b/src/lang/expr.cc
@@ -52,6 +52,14 @@ void Dump(const NodeRef& n) {
   std::cerr << n << "\n";
 }
 
+std::unique_ptr<char[]> ToString(const NodeRef& n) {
+  std::ostringstream s;
+  s << n << "\n";
+  std::unique_ptr<char[]> cc(new char[s.str().length() + 1]);
+  strcpy(cc.get(), s.str().c_str());
+  return cc;
+}
+
 Var var(const std::string& name_hint, Type t) {
   return Var(name_hint, t);
 }