From 8cb61f6d7c742eeedbd834abb171610b8e6e6398 Mon Sep 17 00:00:00 2001 From: An Wang Date: Fri, 4 Feb 2022 10:26:28 -0800 Subject: [PATCH] Revert "[AUTOTVM] Use opt level 3 when extracting tasks (#10065)" This reverts commit 187aeb5fe813c5bffc81e88ba890399ce7d53e1e. --- python/tvm/autotvm/task/relay_integration.py | 15 ++++++--------- 1 file changed, 6 insertions(+), 9 deletions(-) diff --git a/python/tvm/autotvm/task/relay_integration.py b/python/tvm/autotvm/task/relay_integration.py index 04ca333a5ea8..d2861cfe8ad6 100644 --- a/python/tvm/autotvm/task/relay_integration.py +++ b/python/tvm/autotvm/task/relay_integration.py @@ -34,7 +34,7 @@ # TODO(moreau89) find a more elegant way to lower for VTAs -def _lower(mod, target, params, opt_level=3): +def _lower(mod, target, params): """Helper to lower VTA properly.""" # pylint: disable=import-outside-toplevel from tvm import relay @@ -43,19 +43,16 @@ def _lower(mod, target, params, opt_level=3): if hasattr(target, "device_name") and target.device_name == "vta": import vta - with vta.build_config(opt_level=opt_level, disabled_pass={"AlterOpLayout"}): + with vta.build_config(opt_level=3, disabled_pass={"AlterOpLayout"}): mod, _ = relay.optimize(mod, target, params) grc = graph_executor_codegen.GraphExecutorCodegen(None, target) grc.codegen(mod, mod["main"]) return - # Alter op layout code has been written expecting that tuning is applied - # without it, so we disable AlterOpLayout to maintain that behavior. - with tvm.transform.PassContext(opt_level=opt_level, disabled_pass={"AlterOpLayout"}): - compiler = relay.vm.VMCompiler() - if params: - compiler.set_params(params) - compiler.lower(mod, target=target) + compiler = relay.vm.VMCompiler() + if params: + compiler.set_params(params) + compiler.lower(mod, target=target) def extract_from_program(mod, params, target, target_host=None, ops=None):