Support user space installation for rocm

Author: amd-arsuresh

script/README.md

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

script/get-amdclang/customize.py

@@ -31,6 +31,14 @@ def preprocess(i):
             if os.path.isdir(llvm_bin):
                 search_paths.append(llvm_bin)
 
+        # Check custom install prefix (from install-rocm cache)
+        install_prefix = env.get('MLC_ROCM_INSTALL_PREFIX', '')
+        if install_prefix:
+            prefix_opt = os.path.join(install_prefix, 'opt')
+            for p in [os.path.join(prefix_opt, 'rocm', 'llvm', 'bin')] + sorted(glob.glob(os.path.join(prefix_opt, 'rocm-*', 'llvm', 'bin')), reverse=True):
+                if os.path.isdir(p) and p not in search_paths:
+                    search_paths.append(p)
+
         # Also check standard paths
         for p in ['/opt/rocm/llvm/bin'] + sorted(glob.glob('/opt/rocm-*/llvm/bin'), reverse=True):
             if os.path.isdir(p) and p not in search_paths:

script/get-rocm/customize.py

@@ -14,8 +14,18 @@ def preprocess(i):
     file_name = 'rocminfo.exe' if os_info['platform'] == 'windows' else 'rocminfo'
     env['FILE_NAME'] = file_name
 
-    # Build search paths: check /opt/rocm/bin and any versioned /opt/rocm-*/bin
+    # Build search paths: check custom install prefix, then /opt/rocm/bin and versioned /opt/rocm-*/bin
     rocm_paths = []
+
+    # Check custom install prefix (from install-rocm cache)
+    install_prefix = env.get('MLC_ROCM_INSTALL_PREFIX', '')
+    if install_prefix:
+        prefix_opt = os.path.join(install_prefix, 'opt')
+        for p in [os.path.join(prefix_opt, 'rocm', 'bin')] + sorted(glob.glob(os.path.join(prefix_opt, 'rocm-*', 'bin')), reverse=True):
+            if os.path.isdir(p):
+                rocm_paths.append(p)
+
+    # Standard paths
     if os.path.isdir("/opt/rocm/bin"):
         rocm_paths.append("/opt/rocm/bin")
     for p in sorted(glob.glob("/opt/rocm-*/bin"), reverse=True):

script/install-rocm/customize.py

@@ -8,16 +8,47 @@ def preprocess(i):
     if os_info['platform'] == 'windows':
         return {'return': 1, 'error': 'Windows is not supported in this script yet'}
 
+    env = i['env']
+
+    # Determine install prefix:
+    # 1. If explicitly set via input, use that
+    # 2. If sudo is available, default to "/" (installs to /opt/rocm)
+    # 3. Otherwise, use MLC cache directory (no root needed)
+    install_prefix = env.get('MLC_ROCM_INSTALL_PREFIX', '').strip()
+    if install_prefix == '':
+        if env.get('MLC_SUDO_USER', '') == 'yes':
+            install_prefix = '/'
+        else:
+            install_prefix = os.getcwd()
+
+    env['MLC_ROCM_INSTALL_PREFIX'] = install_prefix
+
     return {'return': 0}
 
 
 def postprocess(i):
 
     env = i['env']
 
-    # Check standard and versioned ROCm install paths
+    install_prefix = env.get('MLC_ROCM_INSTALL_PREFIX', '/')
+
+    # Check install prefix paths first, then standard /opt paths
+    search_dirs = []
+
+    # Custom prefix: ROCm installs to <prefix>/opt/rocm-<version>
+    prefix_opt = os.path.join(install_prefix, 'opt')
+    if os.path.isdir(prefix_opt):
+        for p in [os.path.join(prefix_opt, 'rocm', 'bin')] + sorted(glob.glob(os.path.join(prefix_opt, 'rocm-*', 'bin')), reverse=True):
+            if os.path.isdir(p):
+                search_dirs.append(p)
+
+    # Standard paths as fallback
+    for p in ["/opt/rocm/bin"] + sorted(glob.glob("/opt/rocm-*/bin"), reverse=True):
+        if os.path.isdir(p) and p not in search_dirs:
+            search_dirs.append(p)
+
     installed_path = ""
-    for candidate in ["/opt/rocm/bin"] + sorted(glob.glob("/opt/rocm-*/bin"), reverse=True):
+    for candidate in search_dirs:
         if os.path.isfile(os.path.join(candidate, "rocminfo")):
             installed_path = candidate
             break

script/install-rocm/meta.yaml

@@ -14,15 +14,21 @@ tags:
 cache: true
 clean_files: []
 
+# Input mapping
+input_mapping:
+  rocm_install_prefix: MLC_ROCM_INSTALL_PREFIX
+
 # Environment
 env: {}
 new_env_keys:
 - MLC_ROMLC_*
+- MLC_ROCM_INSTALL_PREFIX
 - +PATH
 
 # Dependencies
 deps:
 - tags: detect,os
+- tags: detect,sudo
 
 # Versions
 default_version: 7.2.2

script/install-rocm/run-rhel.sh

@@ -22,9 +22,14 @@ if [[ ${major_version} -ge 7 ]]; then
   test $? -eq 0 || exit 1
   chmod +x /tmp/${runfile_name}
 
-  # Install ROCm via runfile (no driver for CPU-only compatibility)
-  echo "Installing ROCm ${MLC_VERSION} via runfile..."
-  sudo bash /tmp/${runfile_name} deps=install target="/" rocm postrocm
+  # 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
+  else
+    bash /tmp/${runfile_name} deps=install target="${install_prefix}" rocm postrocm
+  fi
   test $? -eq 0 || exit 1
 
   rm -f /tmp/${runfile_name}

script/install-rocm/run-ubuntu.sh

@@ -31,9 +31,14 @@ if [[ ${major_version} -ge 7 ]]; then
   test $? -eq 0 || exit 1
   chmod +x /tmp/${runfile_name}
 
-  # Install ROCm via runfile (no driver for CPU-only compatibility)
-  echo "Installing ROCm ${MLC_VERSION} via runfile..."
-  sudo bash /tmp/${runfile_name} deps=install target="/" rocm postrocm
+  # 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
+  else
+    bash /tmp/${runfile_name} deps=install target="${install_prefix}" rocm postrocm
+  fi
   test $? -eq 0 || exit 1
 
   rm -f /tmp/${runfile_name}