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

---
 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 2a251460e47..468f20952c2 100644
--- 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,
-- 
GitLab