Fixes for 0.5.0 build ported to rocm-main

(cherry picked from commit c23a814)

Author: mrodden

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,

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)