diff --git a/python/tvm/contrib/download.py b/python/tvm/contrib/download.py index bc6d32476eb7..b3ba7efe2a02 100644 --- a/python/tvm/contrib/download.py +++ b/python/tvm/contrib/download.py @@ -120,8 +120,8 @@ def _download_progress(count, block_size, total_size): TEST_DATA_ROOT_PATH = os.path.join(os.path.expanduser('~'), '.tvm_test_data') -if not os.path.exists(TEST_DATA_ROOT_PATH): - os.mkdir(TEST_DATA_ROOT_PATH) +os.makedirs(TEST_DATA_ROOT_PATH, exist_ok=True) + def download_testdata(url, relpath, module=None): """Downloads the test data from the internet.