xla
Update test_pallas_spmd.py for CI
#7637
Merged

Loading