Fix FSDP device mesh initialization on CPU#21816

Yonghui-Lee wants to merge 2 commits into

Lightning-AI:masterLightning-AI/pytorch-lightning:masterfrom

Yonghui-Lee:fsdp-device-mesh-cpu-fixYonghui-Lee/pytorch-lightning:fsdp-device-mesh-cpu-fixCopy head branch name to clipboard