Skip to content

Commit

Permalink
[fbgemm_gpu] Fix validate_binaries.sh for the CPU variant of FBGEMM_GPU
Browse files Browse the repository at this point in the history
- Fix validate_binaries.sh for the CPU variant of FBGEMM_GPU
  • Loading branch information
q10 committed Jan 24, 2025
1 parent e6d5560 commit c5bb731
Showing 1 changed file with 7 additions and 9 deletions.
16 changes: 7 additions & 9 deletions .github/scripts/validate_binaries.sh
Original file line number Diff line number Diff line change
Expand Up @@ -58,15 +58,13 @@ echo "CONDA_ENV: ${CONDA_ENV}"
# shellcheck disable=SC2155
export CONDA_PREFIX=$(conda run -n "${CONDA_ENV}" printenv CONDA_PREFIX)

find / -name *cuda*

if [[ $CUDA_VERSION = cu* ]]; then
# Setting LD_LIBRARY_PATH fixes the runtime error with fbgemm_gpu not
# being able to locate libnvrtc.so
echo "[NOVA] Setting LD_LIBRARY_PATH ..."
conda env config vars set -n ${CONDA_ENV} \
LD_LIBRARY_PATH="/usr/local/lib:/usr/lib64:${CONDA_PREFIX}/lib:${LD_LIBRARY_PATH}"
fi

# Set LD_LIBRARY_PATH to fix the runtime error with fbgemm_gpu not
# being able to locate libnvrtc.so
# NOTE: The order of the entries in LD_LIBRARY_PATH matters
echo "[NOVA] Setting LD_LIBRARY_PATH ..."
conda env config vars set -n ${CONDA_ENV} \
LD_LIBRARY_PATH="${CONDA_PREFIX}/lib:/usr/local/lib:/usr/lib64:${LD_LIBRARY_PATH}"


# install pytorch
Expand Down

0 comments on commit c5bb731

Please sign in to comment.