fix hang of hybrid parallel

PaddlePaddle · Jun 3, 2021 · 23bc09a · 23bc09a
1 parent 4920c47
commit 23bc09a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/paddle/distributed/collective.py b/python/paddle/distributed/collective.py
@@ -263,7 +263,7 @@ def new_group(ranks=None, backend=None):
     else:
         assert False, ("no cuda device found")
     # need to barrier to construct group
-    barrier(gp)
+    barrier()
     return gp