pytorch
076b116a - In ProcessGroupAgent, use non-iostream torch::load()/save(). (#28063)

Commit
5 years ago
In ProcessGroupAgent, use non-iostream torch::load()/save(). (#28063) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/28063 Avoid using the iostream versions of torch::load()/torch::save(), which incur at least one additional full data copy. ghstack-source-id: 92059608 Test Plan: buck test mode/dev-nosan caffe2/test/... Differential Revision: D17945206 fbshipit-source-id: ba24376c13762a28e569530e3b1a939ac6f72f43
Author
Parents
Loading