mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-13 02:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixes for 0.5.0 build ported to rocm-main

Browse Source 
(cherry picked from commit c23a81461192a2b6da3d364076a261714d2dc64f)
This commit is contained in:
Mathew Odden 2025-03-21 18:33:02 -05:00 committed by Mathew Odden
parent 13d88b6340
commit ec4b8ee1ed
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
build/rocm/ci_build
View File

@ -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,

1
build/rocm/tools/build_wheels.py
View File

@ -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)