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 11e7f32
Show file tree
Hide file tree
Showing 2 changed files with 14 additions and 9 deletions.
17 changes: 8 additions & 9 deletions .github/scripts/validate_binaries.sh
Original file line number Diff line number Diff line change
Expand Up @@ -58,15 +58,14 @@ 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
find / -name *libstdc++*

# 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
6 changes: 6 additions & 0 deletions .github/workflows/validate-binaries.yml
Original file line number Diff line number Diff line change
@@ -1,6 +1,12 @@
name: Validate binaries

on:
pull_request:
paths-ignore:
- "docs/*"
- "third_party/*"
- .gitignore
- "*.md"
workflow_call:
inputs:
channel:
Expand Down

0 comments on commit 11e7f32

Please sign in to comment.