Skip to content

Commit 0f37117

Browse files
emilyfertigGoogle-ML-Automation
authored andcommitted
Add multiprocess TPU tests to OSS presubmit.
PiperOrigin-RevId: 834457878
1 parent a284b66 commit 0f37117

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

ci/run_bazel_test_tpu.sh

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -224,7 +224,8 @@ else
224224
//tests:layout_test_tpu \
225225
//tests:pjit_test_tpu \
226226
//tests:python_callback_test_tpu \
227-
//tests:ragged_collective_test_tpu
227+
//tests:ragged_collective_test_tpu \
228+
//tests/multiprocess:tpu_tests
228229

229230
# Store the return value of the second bazel command.
230231
second_bazel_cmd_retval=$?

0 commit comments

Comments
 (0)