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

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

上级 a6b1c4c1
......@@ -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,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册