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

Corresponding build.py updates after 5132a188f7

Browse Source 
PiperOrigin-RevId: 684805296
This commit is contained in:
David Dunleavy 2024-10-11 05:40:50 -07:00 committed by jax authors
parent e58ef1af37
commit ee312afe86
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
build/build.py
View File

@ -285,9 +285,9 @@ def write_bazelrc(*, remote_build,
if enable_cuda:
f.write("build --config=cuda\n")
if use_cuda_nvcc:
f.write("build --config=cuda_nvcc\n")
f.write("build --config=build_cuda_with_nvcc\n")
else:
f.write("build --config=cuda_clang\n")
f.write("build --config=build_cuda_with_clang\n")
f.write(f"build --action_env=CLANG_CUDA_COMPILER_PATH={clang_path}\n")
if not enable_nccl:
f.write("build --config=nonccl\n")