pytorch
4cc474de - [dtensor] support torch.save/load with DTensor (#103106)

Commit
1 year ago
[dtensor] support torch.save/load with DTensor (#103106) This PR actually enables DTensor to be pickable and add tests to test torch.save/load works correctly for DTensor Pull Request resolved: https://github.com/pytorch/pytorch/pull/103106 Approved by: https://github.com/kumpera
Author
Committer
Parents
Loading