Update distributed_c10d.py

Signed-off-by: Jagadish Krishnamoorthy <jagdish.krishna@gmail.com>
pytorch · Dec 23, 2020 · a37fb05 · a37fb05
1 parent edc212e
commit a37fb05
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch/distributed/distributed_c10d.py b/torch/distributed/distributed_c10d.py
@@ -2398,7 +2398,7 @@ def barrier(group=GroupMember.WORLD,
 
     opts = BarrierOptions()
     if device_ids is not None:
-        if get_backend(group) !=  Backend.NCCL:
+        if get_backend(group) != Backend.NCCL:
             raise RuntimeError("Function argument device_ids not supported "
                                "for the selected backend {}".format(get_backend(group)))
         if isinstance(device_ids, list):