Merge pull request #2705 from q10/bm/fix-script

[fbgemm_gpu] Fix validate_binaries.sh for the CPU variant of FBGEMM_GPU

Author: PaulZhang12

.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