[BugFix] Update parallel.py (PaddlePaddle#1905)

ruiwenfan · Apr 7, 2022 · 4ba301b · 4ba301b
1 parent 67d93b1
commit 4ba301b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddlenlp/ops/distributed/parallel.py b/paddlenlp/ops/distributed/parallel.py
@@ -230,7 +230,7 @@ def forward(self, x):
         if self.gather_out is False:
             return output_parallel
 
-        return paddle.distributed.collective._concat(
+        return paddle.distributed.collective._c_concat(
             output_parallel, group=group)