fix bugs: output of splited fc is wrong (#38724)

35213c64 · Roc · GitHub · c0e2b98e · 35213c64
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

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

未找到文件。
--- a/python/paddle/distributed/collective.py
+++ b/python/paddle/distributed/collective.py
@@ -1219,6 +1219,7 @@ def _parallel_linear(x,
            inputs={'X': linear_out},
            outputs={'Out': out},
            attrs={
+                'rank': inner_rank,
                'ring_id': ring_id,
                'nranks': nranks,
                'use_calc_stream': True,