Skip to content

Commit 8f169e7

Browse files
hawkinspGoogle-ML-Automation
authored andcommitted
Disable the paged_attention test on TPU v5p.
This test is failing in CI. PiperOrigin-RevId: 694574616
1 parent 631bcd0 commit 8f169e7

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/pallas/BUILD

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -390,6 +390,9 @@ jax_multiplatform_test(
390390
jax_multiplatform_test(
391391
name = "tpu_paged_attention_kernel_test",
392392
srcs = ["tpu_paged_attention_kernel_test.py"],
393+
disable_configs = [
394+
"tpu_v5p_1x1",
395+
],
393396
enable_backends = ["tpu"],
394397
shard_count = 5,
395398
tags = [

0 commit comments

Comments
 (0)