Make tests multi-process friendly. (#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.
apache · Aug 1, 2019 · 8db483b · 8db483b
1 parent b9544d7
commit 8db483b
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.