xla

6b785fc8 - Pipes attn_logits_soft_cap through multi_queries_paged_attention (#8583)