Skip to content

Commit 4ba301b

Browse files
authored
[BugFix] Update parallel.py (#1905)
1 parent 67d93b1 commit 4ba301b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

paddlenlp/ops/distributed/parallel.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -230,7 +230,7 @@ def forward(self, x):
230230
if self.gather_out is False:
231231
return output_parallel
232232

233-
return paddle.distributed.collective._concat(
233+
return paddle.distributed.collective._c_concat(
234234
output_parallel, group=group)
235235

236236

0 commit comments

Comments
 (0)