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

b533090e · Roc · GitHub · a6b1c4c1 · b533090e
隐藏空白更改
内联并排

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
@@ -1176,6 +1176,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,