debugprint->printir

Author: zhiics

include/tvm/relay/transform.h

@@ -545,7 +545,7 @@ TVM_DLL Pass EtaExpand();
  *
  * \return the pass.
  */
-TVM_DLL Pass DebugPrint();
+TVM_DLL Pass PrintIR();
 
 }  // namespace transform
 

python/tvm/relay/transform.py

@@ -529,7 +529,7 @@ def CanonicalizeCast():
     return _transform.CanonicalizeCast()
 
 
-def DebugPrint():
+def PrintIR():
     """
     Print the IR for a module to help debugging.
 
@@ -538,7 +538,7 @@ def DebugPrint():
     ret : tvm.relay.Pass
         The registered pass that prints the module IR.
     """
-    return _transform.DebugPrint()
+    return _transform.PrintIR()
 
 
 def gradient(expr, mod=None, mode='higher_order'):

src/relay/pass/dump_ir.cc renamed to src/relay/pass/print_ir.cc

@@ -20,7 +20,7 @@
 /*!
  * Copyright (c) 2019 by Contributors
  *
- * \file src/relay/pass/debug_print.cc
+ * \file src/relay/pass/print_ir.cc
  *
  * \brief Print the module IR to help debugging.
  */
@@ -32,17 +32,17 @@ namespace relay {
 
 namespace transform {
 
-Pass DebugPrint() {
+Pass PrintIR() {
   runtime::TypedPackedFunc<Module(Module, PassContext)> pass_func =
     [=](Module m, PassContext pc) {
       LOG(INFO) << "Dumping the module IR: " << std::endl << AsText(m);
       return m;
   };
-  return CreateModulePass(pass_func, 0, "DebugPrint", {});
+  return CreateModulePass(pass_func, 0, "PrintIR", {});
 }
 
-TVM_REGISTER_API("relay._transform.DebugPrint")
-.set_body_typed(DebugPrint);
+TVM_REGISTER_API("relay._transform.PrintIR")
+.set_body_typed(PrintIR);
 
 }  // namespace transform
 

tests/python/relay/test_pass_manager.py

@@ -504,7 +504,7 @@ def expected():
     assert analysis.alpha_equal(zz, zexpected)
 
 
-def test_debug_print():
+def test_print_ir():
     shape = (1, 2, 3)
     tp = relay.TensorType(shape, "float32")
     x = relay.var("x", tp)
@@ -515,7 +515,7 @@ def test_debug_print():
     seq = _transform.Sequential([
         relay.transform.InferType(),
         relay.transform.FoldConstant(),
-        relay.transform.DebugPrint(),
+        relay.transform.PrintIR(),
         relay.transform.DeadCodeElimination()
     ])
 
@@ -568,4 +568,4 @@ def run_pass():
     test_sequential_pass()
     test_sequential_with_scoping()
     test_pass_info()
-    test_debug_print()
+    test_print_ir()