Merge remote-tracking branch 'upstream/main'

Author: jiridanek

jupyter/rocm/tensorflow/ubi9-python-3.12/Dockerfile.rocm

@@ -161,4 +161,9 @@ USER 1001
 
 COPY ${JUPYTER_REUSABLE_UTILS}/usercustomize.pth ${JUPYTER_REUSABLE_UTILS}/monkey_patch_protobuf_6x.py /opt/app-root/lib/python3.12/site-packages/
 
+USER 0
+COPY ${TENSORFLOW_SOURCE_CODE}/utils/link-solibs.sh /tmp/link-solibs.sh
+RUN /tmp/link-solibs.sh && rm /tmp/link-solibs.sh
+USER 1001
+
 WORKDIR /opt/app-root/src

jupyter/rocm/tensorflow/ubi9-python-3.12/utils/link-solibs.sh

@@ -0,0 +1,22 @@
+#!/usr/bin/env bash
+set -Eeuo pipefail
+
+# TensorFlow-ROCm wants to link with `*.so` libraries without version suffixes.
+# This would require us to install the -devel packages, but that's cumbersome with AIPCC bases.
+# Therefore, simply create symlinks to the versioned libraries and (IMPORTANT!) run ldconfig afterwards.
+
+ROCM_PATH=/opt/rocm-6.3.4
+find "$ROCM_PATH/lib" -name '*.so.*' -type f -print0 |
+while IFS= read -r -d '' f; do
+  dir=${f%/*}                       # /opt/rocm-6.3.4/lib  (or sub-dir)
+  bn=${f##*/}                       # libMIOpen.so.1.0.60304
+  base=${bn%%.so*}                  # libMIOpen
+  soname=$base.so                   # libMIOpen.so
+  link=$dir/$soname                 # /opt/rocm-6.3.4/lib/libMIOpen.so
+  [[ -e $link ]] && continue
+  echo "ln -s $bn  →  $link"
+  ln -s "$bn" "$link"
+done
+
+# Run ldconfig to update the cache.
+ldconfig