Skip to content

Fix FSDP device mesh initialization on CPU#21816

Open
Yonghui-Lee wants to merge 2 commits into
Lightning-AI:masterfrom
Yonghui-Lee:fsdp-device-mesh-cpu-fix
Open

Fix FSDP device mesh initialization on CPU#21816
Yonghui-Lee wants to merge 2 commits into
Lightning-AI:masterfrom
Yonghui-Lee:fsdp-device-mesh-cpu-fix