Skip to content

Commit dd614e9

Browse files
committed
Fix
1 parent 04fe843 commit dd614e9

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

python/paddle/distributed/auto_parallel/static/process_group.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -249,9 +249,7 @@ def instantiate(self):
249249
alltoall_tmp = paddle.empty(
250250
shape=[self.nranks, self.nranks], dtype="int32"
251251
)
252-
paddle._legacy_C_ops.alltoall(
253-
alltoall_tmp, 'use_calc_stream', True, 'ring_id', ring_id
254-
)
252+
paddle._C_ops.all_to_all(alltoall_tmp, ring_id)
255253
paddle.device.cuda.synchronize()
256254

257255
if self.nranks > 1:

0 commit comments

Comments
 (0)