Skip to content

Commit ad9e994

Browse files
committed
Restore nvvm-related LD_LIBRARY_PATH, PATH manipulations from main branch.
1 parent 4379bb5 commit ad9e994

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

.github/actions/fetch_ctk/action.yml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -151,13 +151,14 @@ runs:
151151
# mimics actual CTK installation
152152
if [[ "${{ inputs.host-platform }}" == linux* ]]; then
153153
CUDA_PATH=$(realpath "./cuda_toolkit")
154-
echo "LD_LIBRARY_PATH=${LD_LIBRARY_PATH:-}:${CUDA_PATH}/lib" >> $GITHUB_ENV
154+
echo "LD_LIBRARY_PATH=${LD_LIBRARY_PATH:-}:${CUDA_PATH}/lib:${CUDA_PATH}/nvvm/lib64" >> $GITHUB_ENV
155155
elif [[ "${{ inputs.host-platform }}" == win* ]]; then
156156
function normpath() {
157157
echo "$(echo $(cygpath -w $1) | sed 's/\\/\\\\/g')"
158158
}
159159
CUDA_PATH=$(normpath $(realpath "./cuda_toolkit"))
160160
echo "$(normpath ${CUDA_PATH}/bin)" >> $GITHUB_PATH
161+
echo "$(normpath $CUDA_PATH/nvvm/bin)" >> $GITHUB_PATH
161162
fi
162163
echo "CUDA_PATH=${CUDA_PATH}" >> $GITHUB_ENV
163164
echo "CUDA_HOME=${CUDA_PATH}" >> $GITHUB_ENV

0 commit comments

Comments
 (0)