From ec4b8ee1eda2fab2a2e47b4163f53b178908be08 Mon Sep 17 00:00:00 2001
From: Mathew Odden <mathew.odden@amd.com>
Date: Fri, 21 Mar 2025 18:33:02 -0500
Subject: [PATCH] Fixes for 0.5.0 build ported to rocm-main

(cherry picked from commit c23a81461192a2b6da3d364076a261714d2dc64f)
---
 build/rocm/ci_build              | 3 ++-
 build/rocm/tools/build_wheels.py | 1 +
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/build/rocm/ci_build b/build/rocm/ci_build
index b492c808d..ee2c8698d 100755
--- a/build/rocm/ci_build
+++ b/build/rocm/ci_build
@@ -352,6 +352,7 @@ def main():
             args.xla_source_dir,
             args.rocm_build_job,
             args.rocm_build_num,
+            compiler=args.compiler,
         )
 
     elif args.action == "test":
@@ -364,7 +365,7 @@ def main():
             args.xla_source_dir,
             args.rocm_build_job,
             args.rocm_build_num,
-            args.compiler,
+            compiler=args.compiler,
         )
         dist_docker(
             args.rocm_version,
diff --git a/build/rocm/tools/build_wheels.py b/build/rocm/tools/build_wheels.py
index d040d0b70..139a1fdd8 100644
--- a/build/rocm/tools/build_wheels.py
+++ b/build/rocm/tools/build_wheels.py
@@ -291,6 +291,7 @@ def main():
     print("PYTHON_VERSIONS=%r" % python_versions)
     print("JAX_PATH=%s" % args.jax_path)
     print("XLA_PATH=%s" % args.xla_path)
+    print("COMPILER=%s" % args.compiler)
 
     rocm_path = build_rocm_path(args.rocm_version)