Skip to content

Commit 6e917f2

Browse files
committed
REF: Setup CUDA component cache in TMP directory
Prevent collision with components already installed to the CUDA_PATH
1 parent 1016411 commit 6e917f2

File tree

1 file changed

+11
-7
lines changed

1 file changed

+11
-7
lines changed

.github/actions/fetch_ctk/action.yml

+11-7
Original file line numberDiff line numberDiff line change
@@ -50,9 +50,9 @@ runs:
5050
if: ${{ steps.ctk-get-cache.outputs.cache-hit != 'true' }}
5151
shell: bash --noprofile --norc -xeuo pipefail {0}
5252
run: |
53-
CUDA_PATH="./cuda_toolkit"
54-
rm -f $CUDA_PATH
55-
mkdir $CUDA_PATH
53+
CACHE_TMP_DIR="./cache_tmp_dir"
54+
rm -f $CACHE_TMP_DIR
55+
mkdir $CACHE_TMP_DIR
5656
5757
# The binary archives (redist) are guaranteed to be updated as part of the release posting.
5858
CTK_BASE_URL="https://developer.download.nvidia.com/compute/cuda/redist/"
@@ -64,14 +64,14 @@ runs:
6464
CTK_SUBDIR="linux-sbsa"
6565
fi
6666
function extract() {
67-
tar -xvf $1 -C $CUDA_PATH --strip-components=1
67+
tar -xvf $1 -C $CACHE_TMP_DIR --strip-components=1
6868
}
6969
elif [[ "${{ inputs.host-platform }}" == "win-64" ]]; then
7070
CTK_SUBDIR="windows-x86_64"
7171
function extract() {
7272
_TEMP_DIR_=$(mktemp -d)
7373
unzip $1 -d $_TEMP_DIR_
74-
cp -r $_TEMP_DIR_/*/* $CUDA_PATH
74+
cp -r $_TEMP_DIR_/*/* $CACHE_TMP_DIR
7575
rm -rf $_TEMP_DIR_
7676
}
7777
fi
@@ -100,11 +100,15 @@ runs:
100100
for item in $(echo $CTK_CACHE_COMPONENTS | tr ',' ' '); do
101101
populate_cuda_path "$item"
102102
done
103-
ls -l $CUDA_PATH
103+
ls -l $CACHE_TMP_DIR
104104
105105
# Prepare the cache
106106
# Note: try to escape | and > ...
107-
tar -czvf ${CTK_CACHE_FILENAME} ${CUDA_PATH}
107+
tar -czvf ${CTK_CACHE_FILENAME} ${CACHE_TMP_DIR}
108+
109+
CUDA_PATH="./cuda_toolkit"
110+
mkdir -p $CUDA_PATH
111+
mv $CACHE_TMP_DIR/* $CUDA_PATH
108112
109113
- name: Upload CTK cache
110114
if: ${{ always() &&

0 commit comments

Comments
 (0)