From 35213c6469b0b46448a02f76da512c57906a2a83 Mon Sep 17 00:00:00 2001
From: Roc <30228238+sljlp@users.noreply.github.com>
Date: Thu, 6 Jan 2022 10:54:23 +0800
Subject: [PATCH] fix bugs: output of splited fc is wrong (#38724)

---
 python/paddle/distributed/collective.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/python/paddle/distributed/collective.py b/python/paddle/distributed/collective.py
index 14a411ae253..3731332d1e7 100644
--- 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,
-- 
GitLab