Skip to content

Commit 31e61ae

Browse files
emilyfertigGoogle-ML-Automation
authored andcommitted
Disable failing multiprocess array_test on TPU.
PiperOrigin-RevId: 841872712
1 parent c5c8af2 commit 31e61ae

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

ci/run_bazel_test_tpu.sh

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -73,6 +73,9 @@ echo "Running Bazel TPU tests..."
7373
# commands below.
7474
set +e
7575

76+
# TODO(emilyaf): Debug and re-enable this test.
77+
IGNORE_TESTS_MULTIACCELERATOR="-//tests/multiprocess:array_test_tpu"
78+
7679
if [[ "$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=$?
@@ -224,7 +228,8 @@ else
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=$?

0 commit comments

Comments
 (0)