Browse Source

fix remove graph error

tags/v1.3.0
chuxing 3 years ago
parent
commit
10f777d0d4
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/graph/manager/graph_manager.cc

+ 1
- 1
ge/graph/manager/graph_manager.cc View File

@@ -2897,9 +2897,9 @@ void GraphManager::RunThread(GraphManager *graph_manager) {
graph_manager->graph_executor_.SetTrainFlag(graph_manager->options_.train_graph_flag); 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(), ret = graph_manager->graph_executor_.ExecuteGraphAsync(args.graph_id, args.graph_node->GetGeRootModel(),
args.input_tensor); args.input_tensor);
args.graph_node->SetRunFlag(false);
if (ret != SUCCESS) { if (ret != SUCCESS) {
ReturnError(graph_manager, args.callback, ret, "ExecuteGraphAsync failed, thread exit."); ReturnError(graph_manager, args.callback, ret, "ExecuteGraphAsync failed, thread exit.");
args.graph_node->Unlock(); args.graph_node->Unlock();


Loading…
Cancel
Save