diff --git a/manywheel/build_common.sh b/manywheel/build_common.sh index 98c926454..cc56e695a 100644 --- a/manywheel/build_common.sh +++ b/manywheel/build_common.sh @@ -281,7 +281,7 @@ replace_needed_sofiles() { patchedname=$3 if [[ "$origname" != "$patchedname" ]] || [[ "$DESIRED_CUDA" == *"rocm"* ]]; then set +e - origname=$($PATCHELF_BIN --print-needed $sofile | grep "$origname*") + origname=$($PATCHELF_BIN --print-needed $sofile | grep "$origname.*") ERRCODE=$? set -e if [ "$ERRCODE" -eq "0" ]; then diff --git a/manywheel/build_libtorch.sh b/manywheel/build_libtorch.sh index c4dd9aedb..855d4bcc8 100644 --- a/manywheel/build_libtorch.sh +++ b/manywheel/build_libtorch.sh @@ -303,7 +303,7 @@ for pkg in /$LIBTORCH_HOUSE_DIR/libtorch*.zip; do patchedname=${patched[i]} if [[ "$origname" != "$patchedname" ]] || [[ "$DESIRED_CUDA" == *"rocm"* ]]; then set +e - origname=$($PATCHELF_BIN --print-needed $sofile | grep "$origname*") + origname=$($PATCHELF_BIN --print-needed $sofile | grep "$origname.*") ERRCODE=$? set -e if [ "$ERRCODE" -eq "0" ]; then diff --git a/manywheel/build_rocm.sh b/manywheel/build_rocm.sh index b452a2a9e..198237a99 100755 --- a/manywheel/build_rocm.sh +++ b/manywheel/build_rocm.sh @@ -114,7 +114,11 @@ elif [[ "$OS_NAME" == *"Ubuntu"* ]]; then LIBGOMP_PATH="/usr/lib/x86_64-linux-gnu/libgomp.so.1" LIBNUMA_PATH="/usr/lib/x86_64-linux-gnu/libnuma.so.1" LIBELF_PATH="/usr/lib/x86_64-linux-gnu/libelf.so.1" - LIBTINFO_PATH="/lib/x86_64-linux-gnu/libtinfo.so.5" + if [[ $ROCM_INT -ge 50300 ]]; then + LIBTINFO_PATH="/lib/x86_64-linux-gnu/libtinfo.so.6" + else + LIBTINFO_PATH="/lib/x86_64-linux-gnu/libtinfo.so.5" + fi LIBDRM_PATH="/usr/lib/x86_64-linux-gnu/libdrm.so.2" LIBDRM_AMDGPU_PATH="/usr/lib/x86_64-linux-gnu/libdrm_amdgpu.so.1" MAYBE_LIB64=lib