[incubator-tvm] branch master updated: [AutoTVM, Relay] Clear compile engine after task extraction (#5724)

[kevinthesun]

This is an automated email from the ASF dual-hosted git repository.

kevinthesun pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git


The following commit(s) were added to refs/heads/master by this push:
     new a642089  [AutoTVM, Relay] Clear compile engine after task extraction 
(#5724)
a642089 is described below

commit a64208910b22b60da5eb312d3a63cffb6e11f747
Author: Wuwei Lin <wu...@apache.org>
AuthorDate: Thu Jun 4 02:47:04 2020 -0400

    [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