File tree Expand file tree Collapse file tree 1 file changed +2
-1
lines changed
.github/actions/fetch_ctk Expand file tree Collapse file tree 1 file changed +2
-1
lines changed Original file line number Diff line number Diff line change @@ -151,13 +151,14 @@ runs:
151
151
# mimics actual CTK installation
152
152
if [[ "${{ inputs.host-platform }}" == linux* ]]; then
153
153
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
155
155
elif [[ "${{ inputs.host-platform }}" == win* ]]; then
156
156
function normpath() {
157
157
echo "$(echo $(cygpath -w $1) | sed 's/\\/\\\\/g')"
158
158
}
159
159
CUDA_PATH=$(normpath $(realpath "./cuda_toolkit"))
160
160
echo "$(normpath ${CUDA_PATH}/bin)" >> $GITHUB_PATH
161
+ echo "$(normpath $CUDA_PATH/nvvm/bin)" >> $GITHUB_PATH
161
162
fi
162
163
echo "CUDA_PATH=${CUDA_PATH}" >> $GITHUB_ENV
163
164
echo "CUDA_HOME=${CUDA_PATH}" >> $GITHUB_ENV
You can’t perform that action at this time.
0 commit comments