[AutoTVM, Relay] Clear compile engine after task extraction (#5724)

apache · Jun 4, 2020 · a642089 · a642089
1 parent 43dcbc6
commit a642089
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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