pytorch

58ed60c2 - Added context manager enabling all futures returned by rpc_async and custom build rpc functions to be automatically waited on (#41807)

Added context manager enabling all futures returned by rpc_async and custom build rpc functions to be automatically waited on (#41807) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/41807 Test Plan: Make sure ci tests pass, including newly written test Reviewed By: mrshenli Differential Revision: D22640839 Pulled By: osandoval-fb fbshipit-source-id: 3ff98d8e8c6e6d08575e307f05b5e159442d7216