pytorch

54c1cf8d - add distributed checkpoint support for custom device (#120201)

add distributed checkpoint support for custom device (#120201) Fixes #120200 Pull Request resolved: https://github.com/pytorch/pytorch/pull/120201 Approved by: https://github.com/fegin, https://github.com/wz337