Skip to content

Commit

Permalink
download fallback config file for search from tophub if it does not e…
Browse files Browse the repository at this point in the history
…xist (#4671)
  • Loading branch information
Xingyu Zhou authored and kevinthesun committed Jan 10, 2020
1 parent d6a23cf commit 06ce76b
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion python/tvm/autotvm/tophub.py
Original file line number Diff line number Diff line change
Expand Up @@ -214,7 +214,11 @@ def load_reference_log(backend, model, workload_name, template_key):

if key not in REFERENCE_LOG_CACHE:
tmp = []
if os.path.isfile(os.path.join(AUTOTVM_TOPHUB_ROOT_PATH, package_name)):
# 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 os.path.isfile(filename): # in case download failed
find = False
inp = None
counts = {}
Expand Down

0 comments on commit 06ce76b

Please sign in to comment.