xla

b6b9c6da - [SPMD][DTensor] introduce xla_distribute_module for DTensor integration (#6683)