jax
b685442a - Disable `tpu_splash_attention_kernel_test` on TPU v7x.

Commit

194 days ago

Disable `tpu_splash_attention_kernel_test` on TPU v7x. It was timing out. https://source.cloud.google.com/results/invocations/d6f7cee9-3f2a-4c40-bc1f-cd7463a797f5 PiperOrigin-RevId: 857286640

References

#34447 - Disable `tpu_splash_attention_kernel_test` on TPU v7x.

Author

mwhittaker

Committer

Google-ML-Automation

Parents

acc95f10

jax b685442a - Disable `tpu_splash_attention_kernel_test` on TPU v7x.

jax
b685442a - Disable `tpu_splash_attention_kernel_test` on TPU v7x.