版本库

查看Git使用指南 如操作版本库需要认证,请使用您的邮箱
ZIP下载
复制
0 rvarm1@fb.com 提交于 1 day前 :[rpc] make handling of FORWARD_AUTOGRAD_REQ in request_callback_impl (#32476) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/32476 This makes the handling of FORWARD_AUTOGRAD_REQ in request_callback nonblocking. Processing this message requires unwrapping the message with autograd information, processing the original message, and sending back the message with autograd information wrapped. This makes the processing the original message nonblocking by grabbing a future to it and marking the parent future as completed when this one completes. ghstack-source-id: 97221251 Test Plan: `test_rpc_spawn.py` and `test_dist_autograd_spawn.py` both pass. Differential Revision: D19509501 fbshipit-source-id: 84ad2f9c5305ed11ed9bb0144b1aaf5f8698cd2b

ihub@pcl.ac.cn 鹏城实验室人工智能研究中心

版权所有:鹏城实验室    粤ICP备18066427号-6    Powerd by 国防科技大学Trustie