!13729 fix TrainOneStepWithLossScaleCell execorder

From: @youui
Reviewed-by: @kisnwang,@guoqi1024
Signed-off-by: @guoqi1024
pull/13729/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit c66b4e4994

@ -541,6 +541,9 @@ void AscendStreamAssign::TrailingTimeOptimizationByReorder(const NotNull<KernelG
vector<CNodePtr> subgraph_cnodes;
while (pos < cnode_ptr_list.end() && AnfAlgo::GetGraphId((*pos).get()) == subgraph_id) {
if (AnfAlgo::GetCNodeName(*pos) == kLabelGotoOpName) {
break;
}
if (*pos != first_output_ptr) {
subgraph_cnodes.emplace_back(*pos);
} else {

Loading…
Cancel
Save