diff --git a/manywheel/build_common.sh b/manywheel/build_common.sh index 2a6f37ec4..d68d9a323 100644 --- a/manywheel/build_common.sh +++ b/manywheel/build_common.sh @@ -25,6 +25,8 @@ retry () { OS_NAME=$(awk -F= '/^NAME/{print $2}' /etc/os-release) if [[ "$OS_NAME" == *"CentOS Linux"* ]]; then retry yum install -q -y zip openssl +elif [[ "$OS_NAME" == *"AlmaLinux"* ]]; then + retry yum install -q -y zip openssl elif [[ "$OS_NAME" == *"Red Hat Enterprise Linux"* ]]; then retry dnf install -q -y zip openssl elif [[ "$OS_NAME" == *"Ubuntu"* ]]; then diff --git a/manywheel/build_cpu.sh b/manywheel/build_cpu.sh index 24c95f14e..9d982bd30 100755 --- a/manywheel/build_cpu.sh +++ b/manywheel/build_cpu.sh @@ -31,6 +31,8 @@ if [[ "$OS_NAME" == *"CentOS Linux"* ]]; then LIBGOMP_PATH="/usr/lib64/libgomp.so.1" elif [[ "$OS_NAME" == *"Red Hat Enterprise Linux"* ]]; then LIBGOMP_PATH="/usr/lib64/libgomp.so.1" +elif [[ "$OS_NAME" == *"AlmaLinux"* ]]; then + LIBGOMP_PATH="/usr/lib64/libgomp.so.1" elif [[ "$OS_NAME" == *"Ubuntu"* ]]; then if [[ "$(uname -m)" == "s390x" ]]; then LIBGOMP_PATH="/usr/lib/s390x-linux-gnu/libgomp.so.1" diff --git a/manywheel/build_cuda.sh b/manywheel/build_cuda.sh index bc64a5f5d..54a31d521 100644 --- a/manywheel/build_cuda.sh +++ b/manywheel/build_cuda.sh @@ -113,6 +113,8 @@ mkdir -p "$PYTORCH_FINAL_PACKAGE_DIR" || true OS_NAME=$(awk -F= '/^NAME/{print $2}' /etc/os-release) if [[ "$OS_NAME" == *"CentOS Linux"* ]]; then LIBGOMP_PATH="/usr/lib64/libgomp.so.1" +elif [[ "$OS_NAME" == *"AlmaLinux"* ]]; then + LIBGOMP_PATH="/usr/lib64/libgomp.so.1" elif [[ "$OS_NAME" == *"Red Hat Enterprise Linux"* ]]; then LIBGOMP_PATH="/usr/lib64/libgomp.so.1" elif [[ "$OS_NAME" == *"Ubuntu"* ]]; then diff --git a/manywheel/build_libtorch.sh b/manywheel/build_libtorch.sh index 2436d5b10..ea11f0c51 100644 --- a/manywheel/build_libtorch.sh +++ b/manywheel/build_libtorch.sh @@ -24,6 +24,8 @@ retry () { OS_NAME=`awk -F= '/^NAME/{print $2}' /etc/os-release` if [[ "$OS_NAME" == *"CentOS Linux"* ]]; then retry yum install -q -y zip openssl +elif [[ "$OS_NAME" == *"AlmaLinux"* ]]; then + retry yum install -q -y zip openssl elif [[ "$OS_NAME" == *"Red Hat Enterprise Linux"* ]]; then retry dnf install -q -y zip openssl elif [[ "$OS_NAME" == *"Ubuntu"* ]]; then @@ -278,7 +280,7 @@ for pkg in /$LIBTORCH_HOUSE_DIR/libtorch*.zip; do if [[ "$filepath" != "$destpath" ]]; then cp $filepath $destpath fi - + if [[ "$DESIRED_CUDA" == *"rocm"* ]]; then patchedpath=$(fname_without_so_number $destpath) else @@ -299,7 +301,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