已合并
[fix][fully_shard] adjust prefetch, unshard position #707
MengXY107创建于 5月28日
[fix][fully_shard] adjust prefetch, unshard position #707
已合并
Pull Request已成功合入, 合并人@MindSpore-Bot
(感谢 MengXY107 的贡献)5月28日 添加了label:mindspore-cla/yes
5月28日 添加了label:no-pass-all-review
MindSpore-Bot
5月28日 评论:
5月28日 评论:
此处折叠了103条消息 查看更多
司小南(机器人)
6月9日 评论:
6月9日 评论:
| Project Name | Build_Stage | Build Result | Details |
|---|---|---|---|
| Hyper-parallel_Atomgit_Gate | - | ✅ SUCCESS | 3726 |


6月9日 删除了label:ci-pipeline-running
6月9日 添加了label:ci-pipeline-passed
6月9日 合入了pull request,合并节点 SHA:71e8ee29760faa62ecf200c761aa94565b26ff96
What type of PR is this?
/kind bug
What does this PR do / why do we need it:
调整fully_shard中forward_pre_hook, backward_pre_hook中prefetch和unshard的顺序。
当前如果先prefetch再unshard。由于再一条流上,对于第零层来说,会影响第零层的计算和第一层的参数的预取。
Which issue(s) this PR fixes:
Fixes #11
Test Plan and Test result:What scenarios were tested, and what were the verification results(Function, performance, reliability, etc.):
Self-checklist:(请自检,在[ ]内打上x,我们将检视你的完成情况,否则会导致pr无法合入)