More cleanup

apache · Sep 4, 2020 · d3b0c22 · d3b0c22
1 parent 613153e
commit d3b0c22
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 11 deletions.
diff --git a/include/tvm/ir/diagnostic.h b/include/tvm/ir/diagnostic.h
@@ -38,11 +38,12 @@
 #include <tvm/support/logging.h>
 
 #include <fstream>
-#include <rang.hpp>
 #include <string>
 #include <utility>
 #include <vector>
 
+#include <rang.hpp>
+
 namespace tvm {
 
 using tvm::parser::SourceMap;

diff --git a/python/tvm/ir/diagnostic.py b/python/tvm/ir/diagnostic.py
@@ -48,7 +48,7 @@ class Diagnostic(Object):
 # Register the diagnostic renderer.
 @tvm._ffi.register_object("DiagnosticRenderer")
 class DiagnosticRenderer(Object):
-    def __init_(self, render_func):
+    def __init__(self, render_func):
          self.__init_handle_by_constructor__(_mk_renderer, render_func)
 
     def render(self, ctx):

diff --git a/src/parser/parser.cc b/src/parser/parser.cc
@@ -679,7 +679,6 @@ class Parser {
         << "there should be precisely one source in the source_map after parsing";
 
     for (auto func : defs.funcs) {
-      std::cout << func.function;
       mod->Add(func.global, func.function, true);
     }
 

diff --git a/tests/python/relay/test_diagnostic.py b/tests/python/relay/test_diagnostic.py
@@ -7,16 +7,13 @@
 from tvm.runtime import Object
 from tvm.ir.diagnostic import get_default_renderer, set_default_renderer
 
-std_out = get_default_renderer()
+std_out = get_default_renderer()()
 
-    def _renderer(diag_ctx):
-        std_out.render(diag_ctx)
-        file.open(....A).write(json)
-        breakpoint()
+def testing_renderer(diag_ctx):
+    std_out.render(diag_ctx)
+    return
 
-    return mk_renderer(_renderer)
-
-set_default_renderer(the_testing_renderer)
+set_default_renderer(testing_renderer)
 
 def test_span_check():
     data = relay.var('data', shape=(10, 1, 1, 1))