Fix bug in alltoall (#34975)

2e9a31eb · lilong12 · GitHub · a1373714 · 2e9a31eb
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

python/paddle/distributed/collective.py python/paddle/distributed/collective.py +1 -1

未找到文件。
--- 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))