From c5bb731aea3d6dc18ea933edd5bd13f828521e73 Mon Sep 17 00:00:00 2001 From: Benson Ma Date: Fri, 24 Jan 2025 11:37:33 -0800 Subject: [PATCH] [fbgemm_gpu] Fix validate_binaries.sh for the CPU variant of FBGEMM_GPU - Fix validate_binaries.sh for the CPU variant of FBGEMM_GPU --- .github/scripts/validate_binaries.sh | 16 +++++++--------- 1 file changed, 7 insertions(+), 9 deletions(-) diff --git a/.github/scripts/validate_binaries.sh b/.github/scripts/validate_binaries.sh index 6750b5b74..f9958a208 100755 --- a/.github/scripts/validate_binaries.sh +++ b/.github/scripts/validate_binaries.sh @@ -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