From a64208910b22b60da5eb312d3a63cffb6e11f747 Mon Sep 17 00:00:00 2001 From: Wuwei Lin Date: Thu, 4 Jun 2020 02:47:04 -0400 Subject: [PATCH] [AutoTVM, Relay] Clear compile engine after task extraction (#5724) --- python/tvm/autotvm/task/relay_integration.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/tvm/autotvm/task/relay_integration.py b/python/tvm/autotvm/task/relay_integration.py index b7d8fac..9751d90 100644 --- a/python/tvm/autotvm/task/relay_integration.py +++ b/python/tvm/autotvm/task/relay_integration.py @@ -137,6 +137,7 @@ def extract_from_multiple_program(mods, params, target, target_host=None, ops=No args=(mod, target, param)) build_thread.start() build_thread.join() + relay.backend.compile_engine.get().clear() logger.disabled = old_state -- 2.7.4