From d32db392cc79f9230be9d2ec40b030b34d3d3af3 Mon Sep 17 00:00:00 2001 From: Masahiro Masuda Date: Sun, 19 Jan 2020 16:42:22 +0900 Subject: [PATCH] minor fix --- src/relay/backend/contrib/dnnl/codegen.cc | 3 +-- tests/python/relay/test_pass_partition_graph.py | 3 +-- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/src/relay/backend/contrib/dnnl/codegen.cc b/src/relay/backend/contrib/dnnl/codegen.cc index 7ca95c3c12543..807714a33bd5e 100644 --- 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 index a244c50872144..69be0e602d439 100644 --- 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()