@@ -29,14 +29,14 @@ if [[ $JAXCI_RUN_BAZEL_TEST_CPU == 1 ]]; then
2929 # machine can take a long time, we skip running them on these platforms.
3030 if [[ $os == " darwin" ]] || ( [[ $os == " linux" ]] && [[ $arch == " aarch64" ]] ); then
3131 echo " Building RBE CPU tests..."
32- check_if_to_run_in_docker bazel --bazelrc=ci/.bazelrc build --config=rbe_cross_compile_${os} _${arch} \
32+ check_if_to_run_in_docker bazel build --config=rbe_cross_compile_${os} _${arch} \
3333 --repo_env=HERMETIC_PYTHON_VERSION=" $JAXCI_HERMETIC_PYTHON_VERSION " \
3434 --override_repository=xla=" ${JAXCI_XLA_GIT_DIR} " \
3535 --test_env=JAX_NUM_GENERATED_CASES=25 \
3636 //tests:cpu_tests //tests:backend_independent_tests
3737 else
3838 echo " Running RBE CPU tests..."
39- check_if_to_run_in_docker bazel --bazelrc=ci/.bazelrc test --config=rbe_${os} _${arch} \
39+ check_if_to_run_in_docker bazel test --config=rbe_${os} _${arch} \
4040 --repo_env=HERMETIC_PYTHON_VERSION=" $JAXCI_HERMETIC_PYTHON_VERSION " \
4141 --override_repository=xla=" ${JAXCI_XLA_GIT_DIR} " \
4242 --test_env=JAX_NUM_GENERATED_CASES=25 \
@@ -54,15 +54,15 @@ if [[ $JAXCI_RUN_BAZEL_TEST_GPU_LOCAL == 1 ]]; then
5454 # Only Linux x86 builds run GPU tests
5555 # Runs non-multiaccelerator tests with one GPU apiece.
5656 # It appears --run_under needs an absolute path.
57- check_if_to_run_in_docker bazel --bazelrc=ci/.bazelrc test --config=ci_linux_x86_64_cuda \
57+ check_if_to_run_in_docker bazel test --config=ci_linux_x86_64_cuda \
5858 --config=non_multiaccelerator_local \
5959 --repo_env=HERMETIC_PYTHON_VERSION=" $JAXCI_HERMETIC_PYTHON_VERSION " \
6060 --run_under " ${JAXCI_JAX_GIT_DIR} /build/parallel_accelerator_execute.sh" \
6161 //tests:gpu_tests //tests:backend_independent_tests //tests/pallas:gpu_tests //tests/pallas:backend_independent_tests
6262 echo " Finished running non-multiaccelerator tests..."
6363
6464 # Runs multiaccelerator tests with all GPUs.
65- check_if_to_run_in_docker bazel --bazelrc=ci/.bazelrc test --config=ci_linux_x86_64_cuda \
65+ check_if_to_run_in_docker bazel test --config=ci_linux_x86_64_cuda \
6666 --config=multiaccelerator_local \
6767 --repo_env=HERMETIC_PYTHON_VERSION=" $JAXCI_HERMETIC_PYTHON_VERSION " \
6868 //tests:gpu_tests //tests/pallas:gpu_tests
@@ -76,7 +76,7 @@ if [[ $JAXCI_RUN_BAZEL_TEST_GPU_RBE == 1 ]]; then
7676
7777 # Only Linux x86 builds run GPU tests
7878 # Runs non-multiaccelerator tests with one GPU apiece.
79- check_if_to_run_in_docker bazel --bazelrc=ci/.bazelrc test --config=rbe_linux_x86_64_cuda \
79+ check_if_to_run_in_docker bazel test --config=rbe_linux_x86_64_cuda \
8080 --config=non_multiaccelerator \
8181 --repo_env=HERMETIC_PYTHON_VERSION=" $JAXCI_HERMETIC_PYTHON_VERSION " \
8282 --override_repository=xla=" ${JAXCI_XLA_GIT_DIR} " \
0 commit comments