pytorch
d5142c52 - [FSDP]Remove dim_group from device_mesh init (#103218)

Commit
1 year ago
[FSDP]Remove dim_group from device_mesh init (#103218) 1) remove dim_group 2) don't init device_mesh if not using default_pg Pull Request resolved: https://github.com/pytorch/pytorch/pull/103218 Approved by: https://github.com/wanchaol, https://github.com/fduwjj
Author
Committer
Parents
Loading