Skip to content

Commit 7c16560

Browse files
authored
Backport fetch_ctk fixes (NVIDIA#1301)
* backport fetch_ctk fixes * check if the lib subdir exists
1 parent 1d8b02e commit 7c16560

1 file changed

Lines changed: 6 additions & 1 deletion

File tree

.github/actions/fetch_ctk/action.yml

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -123,6 +123,10 @@ runs:
123123
for item in $(echo $CTK_CACHE_COMPONENTS | tr ',' ' '); do
124124
populate_cuda_path "$item"
125125
done
126+
# TODO: check Windows
127+
if [[ "${{ inputs.host-platform }}" == linux* && -d "${CACHE_TMP_DIR}/lib" ]]; then
128+
mv $CACHE_TMP_DIR/lib $CACHE_TMP_DIR/lib64
129+
fi
126130
ls -l $CACHE_TMP_DIR
127131
128132
# Prepare the cache
@@ -176,7 +180,8 @@ runs:
176180
# mimics actual CTK installation
177181
if [[ "${{ inputs.host-platform }}" == linux* ]]; then
178182
CUDA_PATH=$(realpath "${{ inputs.cuda-path }}")
179-
echo "LD_LIBRARY_PATH=${CUDA_PATH}/lib:${LD_LIBRARY_PATH:-}" >> $GITHUB_ENV
183+
echo "${CUDA_PATH}/bin" >> $GITHUB_PATH
184+
echo "LD_LIBRARY_PATH=${CUDA_PATH}/lib64:${LD_LIBRARY_PATH:-}" >> $GITHUB_ENV
180185
elif [[ "${{ inputs.host-platform }}" == win* ]]; then
181186
function normpath() {
182187
echo "$(echo $(cygpath -w $1) | sed 's/\\/\\\\/g')"

0 commit comments

Comments
 (0)