xla

85664631 - backport #7609: Run paged_attention test with megacore mode only on TPU v4 (#7628)