Update torch/testing/_internal/distributed/distributed_test.py

jaglinux · jeffdaily · web-flow · commit 35d02df14521 · 2020-11-13T10:32:28.000-08:00
Co-authored-by: Jeff Daily &lt;jeff.daily@amd.com&gt;
diff --git a/torch/testing/_internal/distributed/distributed_test.py b/torch/testing/_internal/distributed/distributed_test.py
@@ -364,9 +364,8 @@ def _init_multigpu_helper(self):
             if BACKEND == "nccl":
                 apply_hack_for_nccl()
 
-            """ If rank is lesser than or equal to number of available GPU's
-            then each rank can be mapped to corresponding GPU.
-            """
+            # If rank is lesser than or equal to number of available GPU's
+            # then each rank can be mapped to corresponding GPU.
             nGPUs_per_process = 1
             if world_size > nGPUs:
                nGPUs_per_process = nGPUs // world_size
