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

- Fix validate_binaries.sh for the CPU variant of FBGEMM_GPU
pytorch · Jan 24, 2025 · 46751ab · 46751ab
1 parent e6d5560
commit 46751ab
Showing 1 changed file with 5 additions and 7 deletions.
diff --git a/.github/scripts/validate_binaries.sh b/.github/scripts/validate_binaries.sh
@@ -60,13 +60,11 @@ 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
+# 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}"
 
 
 # install pytorch