try cache

nod-ai · Apr 16, 2024 · c6fd3de · c6fd3de
1 parent 1c22eed
commit c6fd3de
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/.github/workflows/test_iree.yml b/.github/workflows/test_iree.yml
@@ -109,6 +109,7 @@ jobs:
       - name: "Running real weight model tests"
         env:
             IREE_TEST_CONFIG_FILES: iree_tests/configs/config_pytorch_models_cpu_llvm_task.json
+            IREE_TEST_FILES: ~/iree_tests_cache
         run: |
           source ${VENV_DIR}/bin/activate
           pytest iree_tests/pytorch/models -s -n 4 -k real_weights -rpfE --timeout=1200 --retries 2 --retry-delay 5 --durations=0
diff --git a/iree_tests/download_remote_files.py b/iree_tests/download_remote_files.py
@@ -84,6 +84,7 @@ def download_azure_remote_file(test_dir: Path, remote_file: str):
             local_dir_path = test_dir
             local_file_path = test_dir / remote_file_name
         else:
+            cache_location = os.path.expanduser(cache_location)
             local_dir_path = Path(cache_location) / "iree_tests" / relative_dir
             local_file_path = Path(cache_location) / "iree_tests" / relative_dir / remote_file_name
         if check_azure_remote_file_matching_local(blob_properties, local_file_path):