File tree 3 files changed +7
-3
lines changed
3 files changed +7
-3
lines changed Original file line number Diff line number Diff line change @@ -281,7 +281,7 @@ replace_needed_sofiles() {
281
281
patchedname=$3
282
282
if [[ " $origname " != " $patchedname " ]] || [[ " $DESIRED_CUDA " == * " rocm" * ]]; then
283
283
set +e
284
- origname=$( $PATCHELF_BIN --print-needed $sofile | grep " $origname *" )
284
+ origname=$( $PATCHELF_BIN --print-needed $sofile | grep " $origname . *" )
285
285
ERRCODE=$?
286
286
set -e
287
287
if [ " $ERRCODE " -eq " 0" ]; then
Original file line number Diff line number Diff line change @@ -303,7 +303,7 @@ for pkg in /$LIBTORCH_HOUSE_DIR/libtorch*.zip; do
303
303
patchedname=${patched[i]}
304
304
if [[ " $origname " != " $patchedname " ]] || [[ " $DESIRED_CUDA " == * " rocm" * ]]; then
305
305
set +e
306
- origname=$( $PATCHELF_BIN --print-needed $sofile | grep " $origname *" )
306
+ origname=$( $PATCHELF_BIN --print-needed $sofile | grep " $origname . *" )
307
307
ERRCODE=$?
308
308
set -e
309
309
if [ " $ERRCODE " -eq " 0" ]; then
Original file line number Diff line number Diff line change @@ -114,7 +114,11 @@ elif [[ "$OS_NAME" == *"Ubuntu"* ]]; then
114
114
LIBGOMP_PATH=" /usr/lib/x86_64-linux-gnu/libgomp.so.1"
115
115
LIBNUMA_PATH=" /usr/lib/x86_64-linux-gnu/libnuma.so.1"
116
116
LIBELF_PATH=" /usr/lib/x86_64-linux-gnu/libelf.so.1"
117
- LIBTINFO_PATH=" /lib/x86_64-linux-gnu/libtinfo.so.5"
117
+ if [[ $ROCM_INT -ge 50300 ]]; then
118
+ LIBTINFO_PATH=" /lib/x86_64-linux-gnu/libtinfo.so.6"
119
+ else
120
+ LIBTINFO_PATH=" /lib/x86_64-linux-gnu/libtinfo.so.5"
121
+ fi
118
122
LIBDRM_PATH=" /usr/lib/x86_64-linux-gnu/libdrm.so.2"
119
123
LIBDRM_AMDGPU_PATH=" /usr/lib/x86_64-linux-gnu/libdrm_amdgpu.so.1"
120
124
MAYBE_LIB64=lib
You can’t perform that action at this time.
0 commit comments