Skip to content

Commit

Permalink
Update distributed.rst to show that CUDA send/recv on GPU is supported (
Browse files Browse the repository at this point in the history
#65601)

Summary:
Pull Request resolved: #65601

I believe this feature was supported one year ago:
#44921

#Closes: #65525
ghstack-source-id: 138918961

Test Plan: N/A

Reviewed By: pritamdamania87, mingzhe09088

Differential Revision: D31163535

fbshipit-source-id: 9321a0a5137a3e265e2b54bd78730ac28c7acd55
  • Loading branch information
Yi Wang authored and facebook-github-bot committed Sep 24, 2021
1 parent 760aefd commit 7f25c3e
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions docs/source/distributed.rst
Expand Up @@ -25,9 +25,9 @@ MPI supports CUDA only if the implementation used to build PyTorch supports it.
+----------------+-----+-----+-----+-----+-----+-----+
| Device | CPU | GPU | CPU | GPU | CPU | GPU |
+================+=====+=====+=====+=====+=====+=====+
| send |||| ? || |
| send |||| ? || |
+----------------+-----+-----+-----+-----+-----+-----+
| recv |||| ? || |
| recv |||| ? || |
+----------------+-----+-----+-----+-----+-----+-----+
| broadcast |||| ? |||
+----------------+-----+-----+-----+-----+-----+-----+
Expand Down

0 comments on commit 7f25c3e

Please sign in to comment.