From c6fd3deb2e708b5515c2b0f029449d8db8a09372 Mon Sep 17 00:00:00 2001 From: saienduri Date: Tue, 16 Apr 2024 00:49:00 -0700 Subject: [PATCH] try cache --- .github/workflows/test_iree.yml | 1 + iree_tests/download_remote_files.py | 1 + 2 files changed, 2 insertions(+) diff --git a/.github/workflows/test_iree.yml b/.github/workflows/test_iree.yml index cbae37d51..7a7c25173 100644 --- 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 index b2d97e290..4b2949f32 100644 --- 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):