From 7f25c3e666c64fddc43771e97c00c41052494aca Mon Sep 17 00:00:00 2001 From: Yi Wang Date: Fri, 24 Sep 2021 12:26:21 -0700 Subject: [PATCH] Update distributed.rst to show that CUDA send/recv on GPU is supported (#65601) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/65601 I believe this feature was supported one year ago: https://github.com/pytorch/pytorch/pull/44921 #Closes: https://github.com/pytorch/pytorch/issues/65525 ghstack-source-id: 138918961 Test Plan: N/A Reviewed By: pritamdamania87, mingzhe09088 Differential Revision: D31163535 fbshipit-source-id: 9321a0a5137a3e265e2b54bd78730ac28c7acd55 --- docs/source/distributed.rst | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/source/distributed.rst b/docs/source/distributed.rst index c5cd727fa7ea..0beca0be3e40 100644 --- 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 | ✓ | ✓ | ✓ | ? | ✘ | ✓ | +----------------+-----+-----+-----+-----+-----+-----+