Update distributed.rst to show that CUDA send/recv on GPU is supported

#Closes: #65525 Differential Revision: [D31163535](https://our.internmc.facebook.com/intern/diff/D31163535/) [ghstack-poisoned]
pytorch · Sep 24, 2021 · 51888c4 · 51888c4
1 parent d07b2cb
commit 51888c4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs/source/distributed.rst b/docs/source/distributed.rst
@@ -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      | ✓   | ✓   | ✓   | ?   | ✘   | ✓   |
 +----------------+-----+-----+-----+-----+-----+-----+