xla

9a1405d6 - Fix the signature of paged_attention by marking megacore_mode optional (#8615)