Added custom-flags variation for install-llvm

Author: amd-arsuresh

script/README.md

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

script/get-llvm/meta.yaml

@@ -2,8 +2,6 @@ alias: get-llvm
 uid: 99832a103ed04eb8
 automation_alias: script
 automation_uid: 5b4e0237da074764
-
-# Metadata
 name: Detect or install LLVM compiler
 category: Compiler automation
 tags:
@@ -13,12 +11,8 @@ tags:
 - c-compiler
 - cpp-compiler
 - get-llvm
-
-# Cache
 cache: true
 clean_files: []
-
-# Environment
 env:
   MLC_REQUIRE_INSTALL: 'no'
 new_env_keys:
@@ -33,8 +27,6 @@ new_env_keys:
 - + LDFLAGS
 - +MLC_HOST_OS_DEFAULT_INCLUDE_PATH
 - +PATH
-
-# Dependencies
 prehook_deps:
 - enable_if_env:
     MLC_REQUIRE_INSTALL:
@@ -47,8 +39,6 @@ prehook_deps:
   tags: install,llvm
 post_deps:
 - tags: get,compiler-flags
-
-# Variations
 variations:
   from-prebuilt:
     group: install-src
@@ -72,6 +62,7 @@ variations:
   sha.#: {}
   tag.#: {}
   version.#: {}
-
-# Output / debugging
+  custom-flags.#:
+    env:
+      MLC_LLVM_CUSTOM_FLAGS: '#'
 sort: 100

script/install-llvm-src/customize.py

@@ -89,6 +89,12 @@ def preprocess(i):
 
             cmake_cmd = f"""cmake {os.path.join(env["MLC_LLVM_SRC_REPO_PATH"], "llvm")} -GNinja -DCMAKE_BUILD_TYPE={llvm_build_type} -DLLVM_ENABLE_PROJECTS={q}{enable_projects}{q} -DLLVM_ENABLE_RUNTIMES={q}{enable_runtimes}{q} -DCMAKE_INSTALL_PREFIX={q}{install_prefix}{q} -DLLVM_ENABLE_RTTI=ON  -DLLVM_INSTALL_UTILS=ON -DLLVM_TARGETS_TO_BUILD={targets_to_build} {cross_compile_options} {target_triple_string} {compiler_rt_target_triple_string} {extra_cmake_options}"""
 
+            # Append custom C/CXX flags if specified via _custom-flags.# variation
+            custom_flags = env.get('MLC_LLVM_CUSTOM_FLAGS', '').strip()
+            if custom_flags:
+                cmake_cmd += f' -DCMAKE_C_FLAGS={q}{custom_flags}{q}'
+                cmake_cmd += f' -DCMAKE_CXX_FLAGS={q}{custom_flags}{q}'
+
             env['MLC_LLVM_CMAKE_CMD'] = cmake_cmd
 
     need_version = env.get('MLC_VERSION', '')

script/install-llvm-src/meta.yaml

@@ -2,8 +2,6 @@ alias: install-llvm-src
 uid: 2af16e9a6c5f4702
 automation_alias: script
 automation_uid: 5b4e0237da074764
-
-# Metadata
 name: Build LLVM compiler from sources (can take >30 min)
 category: Compiler automation
 tags:
@@ -12,20 +10,14 @@ tags:
 - llvm
 - from.src
 - src-llvm
-
-# Cache
 cache: true
-
-# Environment
 env:
   MLC_GIT_URL: https://github.com/llvm/llvm-project
 new_env_keys:
 - MLC_LLVM_*
 - MLC_GET_DEPENDENT_CACHED_PATH
 - +PATH
 - +C_INCLUDE_PATH
-
-# Input mapping
 input_mapping:
   clean: MLC_CLEAN
   extra_options: MLC_LLVM_EXTRA_CMAKE_OPTIONS
@@ -34,8 +26,6 @@ input_mapping:
   target_triple: MLC_LLVM_TARGET_TRIPLE
   targets: MLC_LLVM_TARGETS_TO_BUILD
   src_path: MLC_LLVM_SRC_REPO_PATH
-
-# Dependencies
 deps:
 - tags: detect,os
 - tags: detect,cpu
@@ -59,7 +49,7 @@ deps:
     MLC_LLVM_INSTALLED_PATH:
     - true
     MLC_LLVM_SRC_REPO_PATH:
-    - on
+    - true
   update_tags_from_env_with_prefix:
     _branch.:
     - MLC_GIT_CHECKOUT
@@ -86,8 +76,6 @@ post_deps:
   update_tags_from_env_with_prefix:
     _path.:
     - MLC_LLVM_INSTALLED_PATH
-
-# Variations
 variations:
   debug:
     env:
@@ -196,7 +184,9 @@ variations:
     default_env:
       MLC_LLVM_TARGETS_TO_BUILD: AArch64
     env:
-      MLC_LLVM_CROSS_COMPILE_FLAGS: -DCMAKE_SYSTEM_NAME=Linux  -DCMAKE_SYSTEM_PROCESSOR=aarch64 -DLLVM_DEFAULT_TARGET_TRIPLE=aarch64-linux-gnu -DCMAKE_Fortran_COMPILER=aarch64-linux-gnu-gfortran-12 -DCMAKE_C_COMPILER=aarch64-linux-gnu-gcc  -DCMAKE_CXX_COMPILER=aarch64-linux-gnu-g++
+      MLC_LLVM_CROSS_COMPILE_FLAGS: -DCMAKE_SYSTEM_NAME=Linux  -DCMAKE_SYSTEM_PROCESSOR=aarch64
+        -DLLVM_DEFAULT_TARGET_TRIPLE=aarch64-linux-gnu -DCMAKE_Fortran_COMPILER=aarch64-linux-gnu-gfortran-12
+        -DCMAKE_C_COMPILER=aarch64-linux-gnu-gcc  -DCMAKE_CXX_COMPILER=aarch64-linux-gnu-g++
   cross-compile-x86-aarch64,flang:
     deps:
     - tags: get,generic-sys-util,_gfortran-12-aarch64-linux-gnu
@@ -335,22 +325,17 @@ variations:
     - full-history
     env:
       MLC_GIT_CHECKOUT_TAG: '#'
-
-# Versions
+  custom-flags.#:
+    env:
+      MLC_LLVM_CUSTOM_FLAGS: '#'
 versions: {}
-
-# Docker / container
 docker:
   skip_run_cmd: true
   pre_run_cmds:
   - mlc pull repo
   mounts:
-    - ${{ MLC_LLVM_SRC_REPO_PATH }}:${{ MLC_LLVM_SRC_REPO_PATH }}
-
-# Output / debugging
+  - ${{ MLC_LLVM_SRC_REPO_PATH }}:${{ MLC_LLVM_SRC_REPO_PATH }}
 sort: 1000
-
-# Tests
 tests:
   run_inputs:
   - version: 20.1.1