Use download,file for downloading rocm file

Author: amd-arsuresh

script/README.md

@@ -1,6 +1,6 @@
 # MLCommons Automation Scripts
 
-*Last updated: 2026-04-23 01:58:32*
+*Last updated: 2026-04-23 02:01:43*
 
 This directory contains automation scripts for MLPerf benchmarks, AI/ML workflows, and development operations.
 

script/install-rocm/customize.py

@@ -1,6 +1,12 @@
 from mlc import utils
 import os
 import glob
+import re
+
+try:
+    from urllib.request import urlopen
+except ImportError:
+    from urllib2 import urlopen
 
 
 def preprocess(i):
@@ -23,6 +29,35 @@ def preprocess(i):
 
     env['MLC_ROCM_INSTALL_PREFIX'] = install_prefix
 
+    # For ROCm 7+, resolve the runfile download URL
+    version = env.get('MLC_VERSION', '')
+    major_version = int(version.split('.')[0]) if version else 0
+
+    if major_version >= 7:
+        os_flavor = env.get('MLC_HOST_OS_FLAVOR', 'ubuntu')
+        os_version = env.get('MLC_HOST_OS_VERSION', '')
+
+        if 'ubuntu' in os_flavor or 'debian' in os_flavor:
+            runfile_base_url = f"https://repo.radeon.com/rocm/installer/rocm-runfile-installer/rocm-rel-{version}/ubuntu/{os_version}/"
+        else:
+            runfile_base_url = f"https://repo.radeon.com/rocm/installer/rocm-runfile-installer/rocm-rel-{version}/rhel/{os_version}/"
+
+        # Fetch directory listing to find the runfile name
+        try:
+            response = urlopen(runfile_base_url)
+            html = response.read().decode('utf-8')
+            match = re.search(r'(rocm-installer[^"]+\.run)', html)
+            if match:
+                runfile_name = match.group(1)
+                env['MLC_DOWNLOAD_URL'] = runfile_base_url + runfile_name
+                env['MLC_DOWNLOAD_FILENAME'] = runfile_name
+                env['MLC_ROCM_RUNFILE_NAME'] = runfile_name
+                env['MLC_ROCM_USE_RUNFILE'] = 'yes'
+            else:
+                return {'return': 1, 'error': f'Could not find ROCm runfile installer at {runfile_base_url}'}
+        except Exception as e:
+            return {'return': 1, 'error': f'Failed to fetch ROCm runfile listing from {runfile_base_url}: {e}'}
+
     return {'return': 0}
 
 

script/install-rocm/meta.yaml

@@ -30,6 +30,18 @@ deps:
 - tags: detect,os
 - tags: detect,sudo
 
+# Prehook dependencies
+prehook_deps:
+- tags: download,file
+  enable_if_env:
+    MLC_ROCM_USE_RUNFILE:
+    - 'yes'
+  env:
+    MLC_DOWNLOAD_FILENAME: <<<MLC_ROCM_RUNFILE_NAME>>>
+  force_env_keys:
+  - MLC_DOWNLOAD_URL
+  - MLC_DOWNLOAD_FILENAME
+
 # Versions
 default_version: 7.2.2
 versions:

script/install-rocm/run-rhel.sh

@@ -6,34 +6,25 @@
 major_version="${MLC_VERSION%%.*}"
 
 if [[ ${major_version} -ge 7 ]]; then
-  # ROCm 7.x: Use runfile installer
-  rhel_ver="${MLC_HOST_OS_VERSION}"
-  runfile_base_url="https://repo.radeon.com/rocm/installer/rocm-runfile-installer/rocm-rel-${MLC_VERSION}/rhel/${rhel_ver}/"
-
-  # Get the runfile name from the directory listing
-  runfile_name=$(wget -q -O - "${runfile_base_url}" | grep -oP 'rocm-installer[^"]+\.run' | head -1)
-  if [[ -z "${runfile_name}" ]]; then
-    echo "ERROR: Could not find ROCm runfile installer at ${runfile_base_url}"
+  # ROCm 7.x: Use runfile installer (downloaded via download,file script)
+  runfile_path="${MLC_DOWNLOAD_DOWNLOADED_PATH}"
+  if [[ -z "${runfile_path}" || ! -f "${runfile_path}" ]]; then
+    echo "ERROR: ROCm runfile not found at ${runfile_path}"
     exit 1
   fi
 
-  echo "Downloading ROCm ${MLC_VERSION} runfile installer: ${runfile_name}"
-  wget -q "${runfile_base_url}${runfile_name}" -O /tmp/${runfile_name}
-  test $? -eq 0 || exit 1
-  chmod +x /tmp/${runfile_name}
+  chmod +x "${runfile_path}"
 
   # Install ROCm via runfile
   install_prefix="${MLC_ROCM_INSTALL_PREFIX:-/}"
   echo "Installing ROCm ${MLC_VERSION} via runfile to target=${install_prefix} ..."
   if [[ "${install_prefix}" == "/" ]]; then
-    sudo bash /tmp/${runfile_name} deps=install target="/" rocm postrocm
+    sudo bash "${runfile_path}" deps=install target="/" rocm postrocm
   else
-    bash /tmp/${runfile_name} deps=install target="${install_prefix}" rocm postrocm
+    bash "${runfile_path}" deps=install target="${install_prefix}" rocm postrocm
   fi
   test $? -eq 0 || exit 1
 
-  rm -f /tmp/${runfile_name}
-
 else
   # ROCm 5.x/6.x: Use package manager method
   mainversion="${MLC_HOST_OS_VERSION%%.*}"

script/install-rocm/run-ubuntu.sh

@@ -16,33 +16,25 @@ fi
 major_version="${MLC_VERSION%%.*}"
 
 if [[ ${major_version} -ge 7 ]]; then
-  # ROCm 7.x: Use runfile installer
-  runfile_base_url="https://repo.radeon.com/rocm/installer/rocm-runfile-installer/rocm-rel-${MLC_VERSION}/ubuntu/${ubuntu_ver}/"
-
-  # Get the runfile name from the directory listing
-  runfile_name=$(wget -q -O - "${runfile_base_url}" | grep -oP 'rocm-installer[^"]+\.run' | head -1)
-  if [[ -z "${runfile_name}" ]]; then
-    echo "ERROR: Could not find ROCm runfile installer at ${runfile_base_url}"
+  # ROCm 7.x: Use runfile installer (downloaded via download,file script)
+  runfile_path="${MLC_DOWNLOAD_DOWNLOADED_PATH}"
+  if [[ -z "${runfile_path}" || ! -f "${runfile_path}" ]]; then
+    echo "ERROR: ROCm runfile not found at ${runfile_path}"
     exit 1
   fi
 
-  echo "Downloading ROCm ${MLC_VERSION} runfile installer: ${runfile_name}"
-  wget -q "${runfile_base_url}${runfile_name}" -O /tmp/${runfile_name}
-  test $? -eq 0 || exit 1
-  chmod +x /tmp/${runfile_name}
+  chmod +x "${runfile_path}"
 
   # Install ROCm via runfile
   install_prefix="${MLC_ROCM_INSTALL_PREFIX:-/}"
   echo "Installing ROCm ${MLC_VERSION} via runfile to target=${install_prefix} ..."
   if [[ "${install_prefix}" == "/" ]]; then
-    sudo bash /tmp/${runfile_name} deps=install target="/" rocm postrocm
+    sudo bash "${runfile_path}" deps=install target="/" rocm postrocm
   else
-    bash /tmp/${runfile_name} deps=install target="${install_prefix}" rocm postrocm
+    bash "${runfile_path}" deps=install target="${install_prefix}" rocm postrocm
   fi
   test $? -eq 0 || exit 1
 
-  rm -f /tmp/${runfile_name}
-
 else
   # ROCm 5.x/6.x: Use package manager method
   sudo mkdir --parents --mode=0755 /etc/apt/keyrings