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

Merge pull request #16682 from hawkinsp:distinct

Browse Source 
PiperOrigin-RevId: 547225300
This commit is contained in:
jax authors 2023-07-11 10:17:05 -07:00
commit 35e0a5fd04
2 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.bazelrc
View File

@ -131,7 +131,6 @@ build:rbe --bes_backend=buildeventservice.googleapis.com
build:rbe --bes_results_url="https://source.cloud.google.com/results/invocations"
build:rbe --bes_timeout=600s
build:rbe --define=EXECUTOR=remote
build:rbe --distinct_host_configuration=false
build:rbe --flaky_test_attempts=3
build:rbe --jobs=200
build:rbe --remote_executor=grpcs://remotebuildexecution.googleapis.com

3
build/build.py
View File

@ -260,10 +260,7 @@ def write_bazelrc(*, python_bin_path, remote_build,
f.write(
f'build:rocm --action_env TF_ROCM_AMDGPU_TARGETS="{rocm_amdgpu_targets}"\n')
if cpu is not None:
f.write("build --distinct_host_configuration=true\n")
f.write(f"build --cpu={cpu}\n")
else:
f.write("build --distinct_host_configuration=false\n")
for o in bazel_options:
f.write(f"build {o}\n")