Support rocm src install

Author: amd-arsuresh

script/README.md

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

script/install-rocm/customize.py

@@ -29,8 +29,37 @@ def preprocess(i):
 
     env['MLC_ROCM_INSTALL_PREFIX'] = install_prefix
 
-    # For ROCm 7+, resolve the runfile download URL
     version = env.get('MLC_VERSION', '')
+
+    # Source build: set up git tag and cmake command, skip runfile logic
+    if env.get('MLC_ROCM_BUILD_FROM_SRC', '') == 'yes':
+        env['MLC_GIT_CHECKOUT_TAG'] = f'rocm-{version}'
+
+        src_path = env.get('MLC_ROCM_LLVM_SRC_PATH', '')
+        build_install_prefix = os.path.join(os.getcwd(), 'install')
+        env['MLC_ROCM_INSTALL_PREFIX'] = build_install_prefix
+
+        targets = 'AMDGPU;X86'
+        projects = 'clang;lld;compiler-rt;clang-tools-extra'
+        runtimes = 'libcxx;libcxxabi;openmp'
+
+        cmake_cmd = (
+            f'cmake {os.path.join(src_path, "llvm")} -GNinja'
+            f' -DCMAKE_BUILD_TYPE=Release'
+            f' -DCMAKE_INSTALL_PREFIX={build_install_prefix}'
+            f' -DLLVM_ENABLE_PROJECTS=\"{projects}\"'
+            f' -DLLVM_ENABLE_RUNTIMES=\"{runtimes}\"'
+            f' -DLLVM_TARGETS_TO_BUILD={targets}'
+            f' -DLLVM_ENABLE_RTTI=ON'
+            f' -DLLVM_INSTALL_UTILS=ON'
+            f' -DLLVM_ENABLE_ASSERTIONS=OFF'
+            f' -DLLVM_ENABLE_PER_TARGET_RUNTIME_DIR=OFF'
+        )
+        env['MLC_ROCM_CMAKE_CMD'] = cmake_cmd
+        i['run_script_input']['script_name'] = 'run-src'
+        return {'return': 0}
+
+    # For ROCm 7+, resolve the runfile download URL
     major_version = int(version.split('.')[0]) if version else 0
 
     if major_version >= 7:
@@ -65,6 +94,33 @@ def postprocess(i):
 
     env = i['env']
 
+    # Source build: look for clang in install dir, create amdclang symlinks
+    if env.get('MLC_ROCM_BUILD_FROM_SRC', '') == 'yes':
+        install_dir = env.get('MLC_ROCM_INSTALL_PREFIX', os.path.join(os.getcwd(), 'install'))
+        bin_dir = os.path.join(install_dir, 'bin')
+
+        clang_path = os.path.join(bin_dir, 'clang')
+        if not os.path.isfile(clang_path):
+            return {'return': 1, 'error': f'ROCm LLVM build failed: clang not found at {clang_path}'}
+
+        # Create amdclang symlinks
+        symlinks = {
+            'amdclang': 'clang',
+            'amdclang++': 'clang++',
+            'amdflang': 'flang-new',
+        }
+        for link_name, target in symlinks.items():
+            link_path = os.path.join(bin_dir, link_name)
+            target_path = os.path.join(bin_dir, target)
+            if os.path.isfile(target_path) and not os.path.exists(link_path):
+                os.symlink(target, link_path)
+                print(f'  Created symlink: {link_name} -> {target}')
+
+        env['MLC_ROMLC_INSTALLED_PATH'] = bin_dir
+        env['MLC_ROMLC_BIN_WITH_PATH'] = os.path.join(bin_dir, 'rocminfo') if os.path.isfile(os.path.join(bin_dir, 'rocminfo')) else clang_path
+        env['+PATH'] = [bin_dir]
+        return {'return': 0}
+
     install_prefix = env.get('MLC_ROCM_INSTALL_PREFIX', '/')
     cur_dir = os.getcwd()
 

script/install-rocm/meta.yaml

@@ -2,35 +2,23 @@ alias: install-rocm
 uid: 9d13f90463ce4545
 automation_alias: script
 automation_uid: 5b4e0237da074764
-
-# Metadata
 category: AI/ML frameworks
 tags:
 - install
 - rocm
 - install-rocm
-
-# Cache
 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
-
-# Prehook dependencies
 prehook_deps:
 - tags: download,file
   extra_cache_tags: rocm,installer,runfile
@@ -43,23 +31,37 @@ prehook_deps:
   force_env_keys:
   - MLC_DOWNLOAD_URL
   - MLC_DOWNLOAD_FILENAME
-
-# Variations
 variations:
   compiler-only:
     env:
       MLC_ROCM_COMPILER_ONLY: 'yes'
-
-# Versions
+  src:
+    env:
+      MLC_ROCM_BUILD_FROM_SRC: 'yes'
+      MLC_GIT_URL: https://github.com/ROCm/llvm-project
+    deps:
+    - tags: get,cmake
+    - tags: get,generic-sys-util,_ninja-build
+    - tags: get,git,repo
+      env:
+        MLC_GIT_CHECKOUT_PATH_ENV_NAME: MLC_ROCM_LLVM_SRC_PATH
+      extra_cache_tags: rocm,llvm,src
+      force_env_keys:
+      - MLC_GIT_*
+      names:
+      - rocm-llvm-src-repo
+      update_tags_from_env_with_prefix:
+        _tag.:
+        - MLC_GIT_CHECKOUT_TAG
+        _repo.:
+        - MLC_GIT_URL
 default_version: 7.2.2
 versions:
-  '7.2.2': {}
-  '7.2.1': {}
-  '6.2.0': {}
-  '6.1.0': {}
-  '5.7.1': {}
-
-# Tests
+  7.2.2: {}
+  7.2.1: {}
+  6.2.0: {}
+  6.1.0: {}
+  5.7.1: {}
 tests:
   run_inputs:
   - variations_list: []

script/install-rocm/run-src.sh

@@ -0,0 +1,37 @@
+#!/bin/bash
+
+# Build ROCm LLVM compiler from source
+
+CUR_DIR=$PWD
+INSTALL_DIR="${MLC_ROCM_INSTALL_PREFIX}"
+
+echo "Building ROCm LLVM from source..."
+echo "  Source: ${MLC_ROCM_LLVM_SRC_PATH}"
+echo "  Install prefix: ${INSTALL_DIR}"
+
+mkdir -p build
+cd build
+test $? -eq 0 || exit $?
+
+echo ""
+echo "Running cmake..."
+echo "${MLC_ROCM_CMAKE_CMD}"
+eval "${MLC_ROCM_CMAKE_CMD}"
+test $? -eq 0 || exit $?
+
+echo ""
+echo "Building with ninja (this may take a while)..."
+ninja
+test $? -eq 0 || exit $?
+
+echo ""
+echo "Installing..."
+ninja install
+test $? -eq 0 || exit $?
+
+# Clean build directory (can be very large)
+cd "${CUR_DIR}"
+rm -rf build
+
+echo ""
+echo "ROCm LLVM built and installed to ${INSTALL_DIR}"