xla

cbe0ca84 - [FSDPv2] Enable auto-wrapping (#6499)

[FSDPv2] Enable auto-wrapping (#6499) Summary: This pull request enables auto-wrapping for FSDPv2 following the original XlaFullyShardedDataParallel design and implementation. Test Plan: python test/spmd/test_fsdp_v2.py -v -k test_fsdp_v2_auto_wrap_