From bb83f94ccb67e68b8361e4dc9f13bd365dea3068 Mon Sep 17 00:00:00 2001 From: Jithun Nair Date: Thu, 27 Jul 2023 22:49:43 +0000 Subject: [PATCH] Set FORCE_RPATH for ROCm --- manywheel/build_rocm.sh | 2 ++ 1 file changed, 2 insertions(+) diff --git a/manywheel/build_rocm.sh b/manywheel/build_rocm.sh index 0c1650f9b..fbbf7d3a6 100755 --- a/manywheel/build_rocm.sh +++ b/manywheel/build_rocm.sh @@ -14,6 +14,8 @@ export USE_STATIC_NCCL=1 export ATEN_STATIC_CUDA=1 export USE_CUDA_STATIC_LINK=1 export INSTALL_TEST=0 # dont install test binaries into site-packages +# Set RPATH instead of RUNPATH when using patchelf to avoid LD_LIBRARY_PATH override +export FORCE_RPATH="--force-rpath" # Keep an array of cmake variables to add to if [[ -z "$CMAKE_ARGS" ]]; then