Skip to content

Commit d72f55e

Browse files
committed
update rbe flag name
1 parent c3c6251 commit d72f55e

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

ci/build_artifacts.sh

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -25,17 +25,17 @@ fi
2525

2626
# Build the jaxlib CPU artifact
2727
if [[ "$JAXCI_BUILD_JAXLIB" == 1 ]]; then
28-
python3 build/build.py jaxlib --ci_mode --use_rbe --python_version=$JAXCI_HERMETIC_PYTHON_VERSION --verbose
28+
python3 build/build.py jaxlib --ci_mode --request_rbe --python_version=$JAXCI_HERMETIC_PYTHON_VERSION --verbose
2929
fi
3030

3131
# Build the jax-cuda-plugin artifact
3232
if [[ "$JAXCI_BUILD_PLUGIN" == 1 ]]; then
33-
python3 build/build.py jax-cuda-plugin --ci_mode --use_rbe --python_version=$JAXCI_HERMETIC_PYTHON_VERSION --verbose
33+
python3 build/build.py jax-cuda-plugin --ci_mode --request_rbe --python_version=$JAXCI_HERMETIC_PYTHON_VERSION --verbose
3434
fi
3535

3636
# Build the jax-cuda-pjrt artifact
3737
if [[ "$JAXCI_BUILD_PJRT" == 1 ]]; then
38-
python3 build/build.py jax-cuda-pjrt --ci_mode --use_rbe --verbose
38+
python3 build/build.py jax-cuda-pjrt --ci_mode --request_rbe --verbose
3939
fi
4040

4141
# After building `jaxlib`, `jaxcuda-plugin`, and `jax-cuda-pjrt`, we run

0 commit comments

Comments
 (0)