minor fix

apache · Jan 19, 2020 · d32db39 · d32db39
1 parent 41ffef7
commit d32db39
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/src/relay/backend/contrib/dnnl/codegen.cc b/src/relay/backend/contrib/dnnl/codegen.cc
@@ -334,8 +334,7 @@ class DNNLModuleCodegen : public CSourceModuleCodegenBase {
  */
 runtime::Module DNNLCompiler(const ObjectRef& ref) {
   DNNLModuleCodegen dnnl;
-  auto ret = dnnl.CreateCSourceModule(ref);
-  return ret;
+  return dnnl.CreateCSourceModule(ref);
 }
 
 TVM_REGISTER_GLOBAL("relay.ext.dnnl").set_body_typed(DNNLCompiler);

diff --git a/tests/python/relay/test_pass_partition_graph.py b/tests/python/relay/test_pass_partition_graph.py
@@ -245,7 +245,6 @@ def check_vm_result():
     def check_graph_runtime_result():
         with relay.build_config(opt_level=3, disabled_pass=["AlterOpLayout"]):
             json, lib, param = relay.build(mod, target=target, params=params)
-        # print(json)
         lib = update_lib(lib)
         rt_mod = tvm.contrib.graph_runtime.create(json, lib, ctx)
 
@@ -258,7 +257,7 @@ def check_graph_runtime_result():
 
         tvm.testing.assert_allclose(out.asnumpy(), result, rtol=tol, atol=tol)
 
-    # check_vm_result()
+    check_vm_result()
     check_graph_runtime_result()