wip

pytorch · Jan 8, 2025 · 8e93669 · 8e93669
1 parent 95cbb48
commit 8e93669
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/.github/scripts/install_fbgemm.sh b/.github/scripts/install_fbgemm.sh
@@ -10,14 +10,16 @@ echo "CHANNEL: ${CHANNEL}"
 echo "CONDA_ENV: ${CONDA_ENV}"
 
 if [[ $CU_VERSION = cu* ]]; then
+    echo "[NOVA] Setting LD_LIBRARY_PATH ..."
     conda env config vars set -p ${CONDA_ENV}  \
         LD_LIBRARY_PATH="/usr/local/lib:${CUDA_HOME}/lib64:${CONDA_ENV}/lib:${LD_LIBRARY_PATH}"
 
-    echo "################################################################################"
-    echo "[NOVA] Finding libnvrtc.so ..."
-    LIB_NVRTC=$(find ${CONDA_ENV} -type f -name libnvrtc.so*)
-    cp ${LIB_NVRTC} ${CONDA_ENV}
-    echo "################################################################################"
+    # echo "################################################################################"
+    # echo "[NOVA] Finding libnvrtc.so ..."
+    # LIB_NVRTC=$(find ${CONDA_ENV} -type f -name libnvrtc.so*)
+    # echo "[NOVA] libnvrtc.so found in ${LIB_NVRTC}; will symlink to "
+    # cp ${LIB_NVRTC} ${CONDA_ENV}
+    # echo "################################################################################"
 fi
 
 if [ "$CHANNEL" = "nightly" ]; then