Skip dnf rocjpeg install when already present via ROCm 7.14 pip wheels

ROCm >= 7.14 distributes the full ROCm stack (including rocJPEG) as pip
wheels (_rocm_sdk_core / _rocm_sdk_devel site-packages) rather than
system RPMs. The rocjpeg-devel, libva-amdgpu and mesa-amdgpu-va-drivers
DNF packages therefore don't exist on the ROCm 7.14 builder image and
the install was failing with "No package rocjpeg available".

Check for librocjpeg.so under /opt/conda (pip-wheel install path) and
/opt/rocm (classic RPM install path) before attempting dnf install. If
already present, skip the install entirely. The existing dnf path is
preserved for ROCm 7.2 and earlier where RPMs are the only source.

Co-authored-by: Cursor <cursoragent@cursor.com>

Author: akashveramd

packaging/install_rocjpeg.sh

@@ -20,6 +20,11 @@ set -euo pipefail
 # test runners). On the plain build image that repo is absent; there we only need
 # to *compile* against rocjpeg.h/librocjpeg.so, so fall back to installing just
 # those with --nodeps (base libva provides the libva.so.2 soname we link).
+#
+# ROCm >= 7.14 distributes the full ROCm stack (including rocJPEG) as pip wheels
+# (_rocm_sdk_core / _rocm_sdk_devel site-packages). In that case librocjpeg.so
+# and rocjpeg.h are already present and the dnf packages don't exist, so we
+# skip the install entirely.
 install_rocjpeg_build_only() {
     dnf install -y --refresh libva
     dnf install -y "dnf-command(download)" >/dev/null 2>&1 || dnf install -y dnf-plugins-core
@@ -28,5 +33,16 @@ install_rocjpeg_build_only() {
     rpm -Uvh --nodeps "${rpm_dir}"/rocjpeg*.rpm
 }
 
-dnf install -y --refresh rocjpeg-devel libva-amdgpu mesa-amdgpu-va-drivers \
-    || install_rocjpeg_build_only
+# Check if librocjpeg is already available (e.g. via ROCm 7.14+ pip wheels).
+if python3 -c "
+import glob, sys
+# _rocm_sdk_core and _rocm_sdk_devel are the pip-wheel-based ROCm installs
+hits = (glob.glob('/opt/conda/**/librocjpeg.so*', recursive=True) +
+        glob.glob('/opt/rocm/lib/librocjpeg.so*'))
+sys.exit(0 if hits else 1)
+" 2>/dev/null; then
+    echo "librocjpeg already present (ROCm pip-wheel install); skipping dnf install."
+else
+    dnf install -y --refresh rocjpeg-devel libva-amdgpu mesa-amdgpu-va-drivers \
+        || install_rocjpeg_build_only
+fi