未验证 提交 35213c64 编写于 作者: R Roc 提交者: GitHub

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

上级 c0e2b98e
...@@ -1219,6 +1219,7 @@ def _parallel_linear(x, ...@@ -1219,6 +1219,7 @@ def _parallel_linear(x,
inputs={'X': linear_out}, inputs={'X': linear_out},
outputs={'Out': out}, outputs={'Out': out},
attrs={ attrs={
'rank': inner_rank,
'ring_id': ring_id, 'ring_id': ring_id,
'nranks': nranks, 'nranks': nranks,
'use_calc_stream': True, 'use_calc_stream': True,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册