pytorch
a9153593 - [future] Undo some recent torch::utils::Future api changes (#36220)

Commit
4 years ago
[future] Undo some recent torch::utils::Future api changes (#36220) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/36220 The torch::utils::Future change from yesterday may have introduced a reference cycle, leading to OOM on PS. This change reverts the lambda capture changes with torch::utils::Future until we can analyze further. ghstack-source-id: 101756106 Test Plan: ctr mobile feed: buck run mode/opt -c=python.package_style=inplace //caffe2/torch/fb/training_toolkit/examples:ctr_mbl_feed_integration -- prod-preset Differential Revision: D20918904 fbshipit-source-id: d637f2370aa72c1765b98f3b9e10eb969a025624
Author
Parents
Loading