rocm: fix dnf install and add ROCM_HOME debug prints

install_rocjpeg.sh: drop non-existent libva-amdgpu and
mesa-amdgpu-va-drivers from the explicit dnf install; rocjpeg-devel's
RPM dependencies pull them in automatically per the official rocJPEG
7.2 package install docs.

repair_wheel.py: remove hardcoded /opt/rocm fallback from
_find_rocjpeg_license() and add debug prints to confirm ROCM_HOME
is set correctly at wheel-repair time.

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

Author: akashveramd

packaging/install_rocjpeg.sh

@@ -39,8 +39,6 @@ sys.exit(0 if hits else 1)
     # librocjpeg's own RPATH resolves it from there — no system install needed.
 else
     # ROCm <=7.2 system RPM path.
-    # Per https://github.com/ROCm/rocJPEG/tree/release/rocm-rel-7.2#libraries:
-    # install the VA-API prerequisites first, then rocjpeg-devel (package
-    # install auto installs remaining dependencies).
-    dnf install -y --refresh libva-amdgpu mesa-amdgpu-va-drivers rocjpeg-devel
+    # rocjpeg-devel's RPM dependencies pull in libva and mesa VA drivers automatically.
+    dnf install -y rocjpeg-devel
 fi

packaging/repair_wheel.py

@@ -136,6 +136,10 @@ def _find_nvjpeg_license():
 def _find_rocjpeg_license():
     """Find rocjpeg's LICENSE file to document the runtime dependency."""
     search_roots = []
+    print(
+        f"[repair_wheel] _find_rocjpeg_license: ROCM_HOME={os.environ.get('ROCM_HOME')!r}",
+        flush=True,
+    )
     for var in ("ROCM_HOME", "ROCM_PATH"):
         if v := os.environ.get(var):
             search_roots.append(Path(v))
@@ -154,8 +158,10 @@ def _find_rocjpeg_license():
             search_roots.append(Path(result.stdout.strip()))
     except Exception:
         pass
-    search_roots.append(Path("/opt/rocm"))
-
+    print(
+        f"[repair_wheel] _find_rocjpeg_license: searching roots={[str(r) for r in search_roots]}",
+        flush=True,
+    )
     for root in search_roots:
         candidate = root / "share" / "doc" / "rocjpeg" / "LICENSE"
         if candidate.is_file():