Skip to content

Commit 73b0993

Browse files
committed
[fbgemm_gpu] Fix validate_binaries.sh for the CPU variant of FBGEMM_GPU
- Fix validate_binaries.sh for the CPU variant of FBGEMM_GPU
1 parent e6d5560 commit 73b0993

File tree

2 files changed

+13
-9
lines changed

2 files changed

+13
-9
lines changed

.github/scripts/validate_binaries.sh

Lines changed: 7 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -58,15 +58,13 @@ echo "CONDA_ENV: ${CONDA_ENV}"
5858
# shellcheck disable=SC2155
5959
export CONDA_PREFIX=$(conda run -n "${CONDA_ENV}" printenv CONDA_PREFIX)
6060

61-
find / -name *cuda*
62-
63-
if [[ $CUDA_VERSION = cu* ]]; then
64-
# Setting LD_LIBRARY_PATH fixes the runtime error with fbgemm_gpu not
65-
# being able to locate libnvrtc.so
66-
echo "[NOVA] Setting LD_LIBRARY_PATH ..."
67-
conda env config vars set -n ${CONDA_ENV} \
68-
LD_LIBRARY_PATH="/usr/local/lib:/usr/lib64:${CONDA_PREFIX}/lib:${LD_LIBRARY_PATH}"
69-
fi
61+
find / -name *libstdc++*
62+
63+
# Setting LD_LIBRARY_PATH fixes the runtime error with fbgemm_gpu not
64+
# being able to locate libnvrtc.so
65+
echo "[NOVA] Setting LD_LIBRARY_PATH ..."
66+
conda env config vars set -n ${CONDA_ENV} \
67+
LD_LIBRARY_PATH="${CONDA_PREFIX}/lib:/usr/local/lib:/usr/lib64:${LD_LIBRARY_PATH}"
7068

7169

7270
# install pytorch

.github/workflows/validate-binaries.yml

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,12 @@
11
name: Validate binaries
22

33
on:
4+
pull_request:
5+
paths-ignore:
6+
- "docs/*"
7+
- "third_party/*"
8+
- .gitignore
9+
- "*.md"
410
workflow_call:
511
inputs:
612
channel:

0 commit comments

Comments
 (0)