add type annotations to torch.nn.parallel._functions

[guilhermeleobas]

Closes gh-49686

[facebook-github-bot]

💊 CI failures summary and remediations

As of commit 34ad00c (more details on the Dr. CI page):

---

• 1/1 failures introduced in this PR

---

🕵️ 1 new failure recognized by patterns

The following CI failures do not appear to be due to upstream breakages:

pytorch_linux_xenial_py3_6_gcc5_4_test (1/1)

Step: "Run tests" (full log | diagnosis details | 🔁 rerun)

Dec 21 19:25:27 [E request_callback_no_python.cpp:636] Received error while processing request type 258: RuntimeError: Can not pickle torch.futures.Future

Dec 21 19:25:27 At:
Dec 21 19:25:27   /opt/conda/lib/python3.6/site-packages/torch/distributed/rpc/internal.py(120): serialize
Dec 21 19:25:27   /opt/conda/lib/python3.6/site-packages/torch/distributed/rpc/internal.py(172): serialize
Dec 21 19:25:27 
Dec 21 19:25:27 [E request_callback_no_python.cpp:636] Received error while processing request type 258: RuntimeError: Can not pickle torch.futures.Future
Dec 21 19:25:27 
Dec 21 19:25:27 At:
Dec 21 19:25:27   /opt/conda/lib/python3.6/site-packages/torch/distributed/rpc/internal.py(120): serialize
Dec 21 19:25:27   /opt/conda/lib/python3.6/site-packages/torch/distributed/rpc/internal.py(172): serialize
Dec 21 19:25:27 
Dec 21 19:25:27 [E request_callback_no_python.cpp:636] Received error while processing request type 258: RuntimeError: Can not pickle torch.futures.Future
Dec 21 19:25:27 
Dec 21 19:25:27 At:
Dec 21 19:25:27   /opt/conda/lib/python3.6/site-packages/torch/distributed/rpc/internal.py(120): serialize
Dec 21 19:25:27   /opt/conda/lib/python3.6/site-packages/torch/distributed/rpc/internal.py(172): serialize
Dec 21 19:25:27 
Dec 21 19:25:27 ok (1.739s)
Dec 21 19:25:28   test_return_future_remote (__main__.ProcessGroupRpcTestWithSpawn) ... RPC was initialized with the PROCESS_GROUP backend which is deprecated and slated to be removed and superseded by the TENSORPIPE backend. It is recommended to migrate to the TENSORPIPE backend.
Dec 21 19:25:29 RPC was initialized with the PROCESS_GROUP backend which is deprecated and slated to be removed and superseded by the TENSORPIPE backend. It is recommended to migrate to the TENSORPIPE backend.
Dec 21 19:25:29 RPC was initialized with the PROCESS_GROUP backend which is deprecated and slated to be removed and superseded by the TENSORPIPE backend. It is recommended to migrate to the TENSORPIPE backend.
Dec 21 19:25:29 RPC was initialized with the PROCESS_GROUP backend which is deprecated and slated to be removed and superseded by the TENSORPIPE backend. It is recommended to migrate to the TENSORPIPE backend.

---

This comment was automatically generated by Dr. CI (expand for details).

Follow this link to opt-out of these comments for your Pull Requests.

Please report bugs/suggestions to the (internal) Dr. CI Users group.

This comment has been revised 6 times.

[rgommers]

LGTM, thanks @guilhermeleobas

[facebook-github-bot]

@zou3519 has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.

[facebook-github-bot]

@zou3519 merged this pull request in 27f0dd3.