[FSDPv2] Move the module to xla device (#6525)
Summary:
This change allows moving module to the xla device during wrapping such that the caller doesn't need to move the module to the xla device.
Test Plan:
python test/spmd/test_fsdp_v2.py -v -k test_fsdp_v2_cpu_model
