New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

bug fix #8730

Merged

wawltor merged 1 commit into PaddlePaddle:develop from FeixLiu:fix_all_gather_bug

Jul 9, 2024

Merged

bug fix #8730

paddlenlp/utils/distributed.py

-Original file line number
+Diff line change
@@ Expand Up / @@ -214,7 +214,7 @@ @@
                     x.reshape_(origin_shape)
             else:
-                distributed.all_gather(output_tensors, tensor)
+                distributed.all_gather(output_tensors, tensor, group=group)
             return output_tensors
@@ Expand Down @@