[release/2.6] Make triton build ROCm version agnostic (#2130)

Fixes
http://rocm-ci.amd.com/job/mainline-pytorch2.6-manylinux-wheels-lw/
 
These are failing 
 
Passed log:
http://rocm-ci.amd.com/job/mainline-pytorch2.6-manylinux-wheels-lw/104/ 
 
Failed log:
http://rocm-ci.amd.com/job/mainline-pytorch2.6-manylinux-wheels-lw/106/ 
 
Error Observed:
14:44:13  +++ find /opt/ -name libamdhip64.so.5
14:44:13  ++ [[ -z '' ]]
14:44:13  ++ echo 'Error: Library file libamdhip64.so.5 is not found.'
14:44:13  Error: Library file libamdhip64.so.5 is not found.
 

Was failing because rocm mainline became 7.0 and not 6.X. Removing the
version calls.
Validation:
http://rocm-ci.amd.com/job/rocm-pytorch-manylinux-wheel-builder-lw/146/

Co-authored-by: Jithun Nair <[email protected]>

Author: ethanwee1

.github/scripts/amd/package_triton_wheel.sh

@@ -1,3 +1,4 @@
+#!/bin/bash
 set -ex
 
 # Set ROCM_HOME isn't available, use ROCM_PATH if set or /opt/rocm
@@ -50,31 +51,16 @@ do
     cp $lib $TRITON_ROCM_DIR/lib/
 done
 
-# Required ROCm libraries
-if [[ "$MAJOR_VERSION" -ge 6 ]]; then
-    libamdhip="libamdhip64.so.6"
-else
-    libamdhip="libamdhip64.so.5"
-fi
-
 # Required ROCm libraries - ROCm 6.0
 ROCM_SO=(
-    "${libamdhip}"
-    "libhsa-runtime64.so.1"
-    "libdrm.so.2"
-    "libdrm_amdgpu.so.1"
+    "libamdhip64.so"
+    "libhsa-runtime64.so"
+    "libdrm.so"
+    "libdrm_amdgpu.so"
+    "libamd_comgr.so"
+    "librocprofiler-register.so"
 )
 
-if [[ $ROCM_INT -ge 60400 ]]; then
-    ROCM_SO+=("libamd_comgr.so.3")
-else
-    ROCM_SO+=("libamd_comgr.so.2")
-fi
-
-if [[ $ROCM_INT -ge 60100 ]]; then
-    ROCM_SO+=("librocprofiler-register.so.0")
-fi
-
 for lib in "${ROCM_SO[@]}"
 do
     file_path=($(find $ROCM_HOME/lib/ -name "$lib")) # First search in lib
@@ -95,10 +81,6 @@ do
     fi
 
     cp $file_path $TRITON_ROCM_DIR/lib
-    # When running locally, and not building a wheel, we need to satisfy shared objects requests that don't look for versions
-    LINKNAME=$(echo $lib | sed -e 's/\.so.*/.so/g')
-    ln -sf $lib $TRITON_ROCM_DIR/lib/$LINKNAME
-
 done
 
 # Copy Include Files

.github/scripts/amd/patch_triton_wheel.sh

@@ -19,15 +19,13 @@ replace_needed_sofiles() {
     find $1 -name '*.so*' -o -name 'ld.lld' | while read sofile; do
         origname=$2
         patchedname=$3
-        if [[ "$origname" != "$patchedname" ]]; then
-            set +e
-            origname=$($PATCHELF_BIN --print-needed $sofile | grep "$origname.*")
-            ERRCODE=$?
-            set -e
-            if [ "$ERRCODE" -eq "0" ]; then
-                echo "patching $sofile entry $origname to $patchedname"
-                $PATCHELF_BIN --replace-needed $origname $patchedname $sofile
-            fi
+        set +e
+        origname=$($PATCHELF_BIN --print-needed $sofile | grep "$origname.*")
+        ERRCODE=$?
+        set -e
+        if [ "$ERRCODE" -eq "0" ]; then
+            echo "patching $sofile entry $origname to $patchedname"
+            $PATCHELF_BIN --replace-needed $origname $patchedname $sofile
         fi
     done
 }