Avoid downloading when TOPHUB_LOCATION is NONE (#5720)

apache · Jun 3, 2020 · 927510a · 927510a
1 parent c1f3b2f
commit 927510a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/python/tvm/autotvm/tophub.py b/python/tvm/autotvm/tophub.py
@@ -216,10 +216,12 @@ def load_reference_log(backend, model, workload_name):
 
     if key not in REFERENCE_LOG_CACHE:
         tmp = []
+        # If TOPHUB_LOCATION is not AUTOTVM_TOPHUB_NONE_LOC,
         # Download the config file from tophub if not exists.
         if not os.path.exists(filename):
             tophub_location = _get_tophub_location()
-            download_package(tophub_location, package_name)
+            if tophub_location != AUTOTVM_TOPHUB_NONE_LOC:
+                download_package(tophub_location, package_name)
         if os.path.isfile(filename): # in case download failed
             find = False
             inp = None