Avoid downloading when TOPHUB_LOCATION is NONE (#5720)
authorlixiaoquan <radioheads@163.com>
Wed, 3 Jun 2020 15:01:10 +0000 (23:01 +0800)
committerGitHub <noreply@github.com>
Wed, 3 Jun 2020 15:01:10 +0000 (08:01 -0700)
python/tvm/autotvm/tophub.py

index a001337..c7c55ed 100644 (file)
@@ -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