Browse Source

!13729 fix TrainOneStepWithLossScaleCell execorder

From: @youui
Reviewed-by: @kisnwang,@guoqi1024
Signed-off-by: @guoqi1024
pull/13729/MERGE
mindspore-ci-bot Gitee 4 years ago
parent
commit
c66b4e4994
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      mindspore/ccsrc/runtime/device/ascend/ascend_stream_assign.cc

+ 3
- 0
mindspore/ccsrc/runtime/device/ascend/ascend_stream_assign.cc View File

@@ -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