Browse Source

Release context in execute end.

tags/v1.3.0
zhaozhixuan 3 years ago
parent
commit
ab7334ed78
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/hybrid/executor/hybrid_model_executor.cc

+ 1
- 1
ge/hybrid/executor/hybrid_model_executor.cc View File

@@ -70,6 +70,7 @@ Status HybridModelExecutor::Execute(HybridModelExecutor::ExecuteArgs &args) {
context_.profiler->Dump(std::cout);
context_.profiler->Reset();
}
root_graph_executor_->ResetContext();

context_.iteration += 1;
if (ret == END_OF_SEQUENCE) {
@@ -125,7 +126,6 @@ Status HybridModelExecutor::ExecuteGraphInternal(HybridModelExecutor::ExecuteArg
args.outputs.clear();
HYBRID_CHK_STATUS_RET(root_graph_executor_->GetOutputs(args.outputs, args.output_desc), "Failed to get outputs");
RECORD_MODEL_EXECUTION_EVENT(&context_, "[GetOutput] End");
root_graph_executor_->ResetContext();
return SUCCESS;
}



Loading…
Cancel
Save