Make tests multi-process friendly. (apache#3683)

This side effect at module import time has a race condition between the "exists" check and the "mkdir" call. The safer thing is to just call mkdir and catch the "already exists" error which is what makedirs does.
neo-ai · Sep 6, 2019 · 6f937f2 · 6f937f2
1 parent cb5702f
commit 6f937f2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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.