File tree Expand file tree Collapse file tree 4 files changed +10
-2
lines changed Expand file tree Collapse file tree 4 files changed +10
-2
lines changed Original file line number Diff line number Diff line change @@ -25,6 +25,8 @@ retry () {
25
25
OS_NAME=$( awk -F= ' /^NAME/{print $2}' /etc/os-release)
26
26
if [[ " $OS_NAME " == * " CentOS Linux" * ]]; then
27
27
retry yum install -q -y zip openssl
28
+ elif [[ " $OS_NAME " == * " AlmaLinux" * ]]; then
29
+ retry yum install -q -y zip openssl
28
30
elif [[ " $OS_NAME " == * " Red Hat Enterprise Linux" * ]]; then
29
31
retry dnf install -q -y zip openssl
30
32
elif [[ " $OS_NAME " == * " Ubuntu" * ]]; then
Original file line number Diff line number Diff line change @@ -31,6 +31,8 @@ if [[ "$OS_NAME" == *"CentOS Linux"* ]]; then
31
31
LIBGOMP_PATH=" /usr/lib64/libgomp.so.1"
32
32
elif [[ " $OS_NAME " == * " Red Hat Enterprise Linux" * ]]; then
33
33
LIBGOMP_PATH=" /usr/lib64/libgomp.so.1"
34
+ elif [[ " $OS_NAME " == * " AlmaLinux" * ]]; then
35
+ LIBGOMP_PATH=" /usr/lib64/libgomp.so.1"
34
36
elif [[ " $OS_NAME " == * " Ubuntu" * ]]; then
35
37
if [[ " $( uname -m) " == " s390x" ]]; then
36
38
LIBGOMP_PATH=" /usr/lib/s390x-linux-gnu/libgomp.so.1"
Original file line number Diff line number Diff line change @@ -113,6 +113,8 @@ mkdir -p "$PYTORCH_FINAL_PACKAGE_DIR" || true
113
113
OS_NAME=$( awk -F= ' /^NAME/{print $2}' /etc/os-release)
114
114
if [[ " $OS_NAME " == * " CentOS Linux" * ]]; then
115
115
LIBGOMP_PATH=" /usr/lib64/libgomp.so.1"
116
+ elif [[ " $OS_NAME " == * " AlmaLinux" * ]]; then
117
+ LIBGOMP_PATH=" /usr/lib64/libgomp.so.1"
116
118
elif [[ " $OS_NAME " == * " Red Hat Enterprise Linux" * ]]; then
117
119
LIBGOMP_PATH=" /usr/lib64/libgomp.so.1"
118
120
elif [[ " $OS_NAME " == * " Ubuntu" * ]]; then
Original file line number Diff line number Diff line change @@ -24,6 +24,8 @@ retry () {
24
24
OS_NAME=` awk -F= ' /^NAME/{print $2}' /etc/os-release`
25
25
if [[ " $OS_NAME " == * " CentOS Linux" * ]]; then
26
26
retry yum install -q -y zip openssl
27
+ elif [[ " $OS_NAME " == * " AlmaLinux" * ]]; then
28
+ retry yum install -q -y zip openssl
27
29
elif [[ " $OS_NAME " == * " Red Hat Enterprise Linux" * ]]; then
28
30
retry dnf install -q -y zip openssl
29
31
elif [[ " $OS_NAME " == * " Ubuntu" * ]]; then
@@ -278,7 +280,7 @@ for pkg in /$LIBTORCH_HOUSE_DIR/libtorch*.zip; do
278
280
if [[ " $filepath " != " $destpath " ]]; then
279
281
cp $filepath $destpath
280
282
fi
281
-
283
+
282
284
if [[ " $DESIRED_CUDA " == * " rocm" * ]]; then
283
285
patchedpath=$( fname_without_so_number $destpath )
284
286
else
@@ -299,7 +301,7 @@ for pkg in /$LIBTORCH_HOUSE_DIR/libtorch*.zip; do
299
301
patchedname=${patched[i]}
300
302
if [[ " $origname " != " $patchedname " ]] || [[ " $DESIRED_CUDA " == * " rocm" * ]]; then
301
303
set +e
302
- origname=$( $PATCHELF_BIN --print-needed $sofile | grep " $origname .*" )
304
+ origname=$( $PATCHELF_BIN --print-needed $sofile | grep " $origname .*" )
303
305
ERRCODE=$?
304
306
set -e
305
307
if [ " $ERRCODE " -eq " 0" ]; then
You can’t perform that action at this time.
0 commit comments