fix remove graph error

4 years ago · 10f777d0d4
parent 9edd57eef2
commit 10f777d0d4
1 changed files with 1 additions and 1 deletions
--- a/ge/graph/manager/graph_manager.cc
+++ b/ge/graph/manager/graph_manager.cc
@ -2897,9 +2897,9 @@ void GraphManager::RunThread(GraphManager *graph_manager) {
      graph_manager->graph_executor_.SetTrainFlag(graph_manager->options_.train_graph_flag);
    }

+    args.graph_node->SetRunFlag(false);
    ret = graph_manager->graph_executor_.ExecuteGraphAsync(args.graph_id, args.graph_node->GetGeRootModel(),
                                                           args.input_tensor);
-    args.graph_node->SetRunFlag(false);
    if (ret != SUCCESS) {
      ReturnError(graph_manager, args.callback, ret, "ExecuteGraphAsync failed, thread exit.");
      args.graph_node->Unlock();