xla

b84f7980 - Add fsdp tests back to GPU CI (#5674)

Add fsdp tests back to GPU CI (#5674) * use manfei's change * use the new flag * reduce data size * reduce batch size * keep reducing batch size to 64 * remove comments