File tree Expand file tree Collapse file tree 1 file changed +7
-2
lines changed
Expand file tree Collapse file tree 1 file changed +7
-2
lines changed Original file line number Diff line number Diff line change @@ -73,6 +73,9 @@ echo "Running Bazel TPU tests..."
7373# commands below.
7474set +e
7575
76+ # TODO(emilyaf): Debug and re-enable this test.
77+ IGNORE_TESTS_MULTIACCELERATOR=" -//tests/multiprocess:array_test_tpu"
78+
7679if [[ " $JAXCI_RUN_FULL_TPU_TEST_SUITE " == " 1" ]]; then
7780 # We're deselecting all Pallas TPU tests in the oldest libtpu build. Mosaic
7881 # TPU does not guarantee anything about forward compatibility (unless
@@ -142,7 +145,8 @@ if [[ "$JAXCI_RUN_FULL_TPU_TEST_SUITE" == "1" ]]; then
142145 //tests:tpu_tests \
143146 //tests/pallas:tpu_tests \
144147 //tests/pallas:tpu_pallas_test_tpu \
145- //tests/multiprocess:tpu_tests
148+ //tests/multiprocess:tpu_tests \
149+ $IGNORE_TESTS_MULTIACCELERATOR
146150
147151 # Store the return value of the second bazel command.
148152 second_bazel_cmd_retval=$?
224228 //tests:pjit_test_tpu \
225229 //tests:python_callback_test_tpu \
226230 //tests:ragged_collective_test_tpu \
227- //tests/multiprocess:tpu_tests
231+ //tests/multiprocess:tpu_tests \
232+ $IGNORE_TESTS_MULTIACCELERATOR
228233
229234 # Store the return value of the second bazel command.
230235 second_bazel_cmd_retval=$?
You can’t perform that action at this time.
0 commit comments