File tree 1 file changed +11
-7
lines changed
.github/actions/fetch_ctk
1 file changed +11
-7
lines changed Original file line number Diff line number Diff line change 50
50
if : ${{ steps.ctk-get-cache.outputs.cache-hit != 'true' }}
51
51
shell : bash --noprofile --norc -xeuo pipefail {0}
52
52
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
56
56
57
57
# The binary archives (redist) are guaranteed to be updated as part of the release posting.
58
58
CTK_BASE_URL="https://developer.download.nvidia.com/compute/cuda/redist/"
@@ -64,14 +64,14 @@ runs:
64
64
CTK_SUBDIR="linux-sbsa"
65
65
fi
66
66
function extract() {
67
- tar -xvf $1 -C $CUDA_PATH --strip-components=1
67
+ tar -xvf $1 -C $CACHE_TMP_DIR --strip-components=1
68
68
}
69
69
elif [[ "${{ inputs.host-platform }}" == "win-64" ]]; then
70
70
CTK_SUBDIR="windows-x86_64"
71
71
function extract() {
72
72
_TEMP_DIR_=$(mktemp -d)
73
73
unzip $1 -d $_TEMP_DIR_
74
- cp -r $_TEMP_DIR_/*/* $CUDA_PATH
74
+ cp -r $_TEMP_DIR_/*/* $CACHE_TMP_DIR
75
75
rm -rf $_TEMP_DIR_
76
76
}
77
77
fi
@@ -100,11 +100,15 @@ runs:
100
100
for item in $(echo $CTK_CACHE_COMPONENTS | tr ',' ' '); do
101
101
populate_cuda_path "$item"
102
102
done
103
- ls -l $CUDA_PATH
103
+ ls -l $CACHE_TMP_DIR
104
104
105
105
# Prepare the cache
106
106
# 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
108
112
109
113
- name : Upload CTK cache
110
114
if : ${{ always() &&
You can’t perform that action at this time.
0 commit comments