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

Pass bazel options to requirements_update and requirements_nightly_update commands

Browse Source
This commit is contained in:
vfdev-5 2024-08-16 11:47:21 +02:00
parent 527a4b8818
commit 12e8bf4525
1 changed files with 6 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
build/build.py
View File

@ -622,20 +622,17 @@ def main():
python_version=python_version,
)
if args.requirements_update:
if args.requirements_update or args.requirements_nightly_update:
if args.requirements_update:
task = "//build:requirements.update"
else: # args.requirements_nightly_update
task = "//build:requirements_nightly.update"
update_command = ([bazel_path] + args.bazel_startup_options +
["run", "--verbose_failures=true", "//build:requirements.update"])
["run", "--verbose_failures=true", task, *args.bazel_options])
print(" ".join(update_command))
shell(update_command)
return
if args.requirements_nightly_update:
update_nightly_command = ([bazel_path] + args.bazel_startup_options +
["run", "--verbose_failures=true", "//build:requirements_nightly.update"])
print(" ".join(update_nightly_command))
shell(update_nightly_command)
return
if args.configure_only:
return