xla

Update numerical verification for SPMD Linear checkpointing

#9113

Closed