From 2e9a31eb5da2a5b4415c87cd8d1400a1e030c1bb Mon Sep 17 00:00:00 2001 From: lilong12 Date: Wed, 18 Aug 2021 11:18:58 +0800 Subject: [PATCH] Fix bug in alltoall (#34975) --- python/paddle/distributed/collective.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/distributed/collective.py b/python/paddle/distributed/collective.py index e5dfb34f243..70e16d67fb9 100644 --- a/python/paddle/distributed/collective.py +++ b/python/paddle/distributed/collective.py @@ -1524,7 +1524,7 @@ def alltoall(in_tensor_list, out_tensor_list, group=None, use_calc_stream=True): inputs={'X': [temp]}, outputs={'Out': [out]}, attrs={ - 'ring_id': group, + 'ring_id': ring_id, 'use_calc_stream': use_calc_stream, }) out_tensor_list.extend(paddle.split(out, nranks, 0)) -- GitLab