|
|
|
@@ -1128,8 +1128,8 @@ Status GraphManager::InnerRunGraph(GraphNodePtr &graph_node, const GraphId &grap |
|
|
|
return SUCCESS; |
|
|
|
} |
|
|
|
|
|
|
|
Status RunGraphWithStreamAsync(const GraphId &graph_id, const std::vector<GeTensor> &inputs, |
|
|
|
std::vector<GeTensor> &outputs, rtStream_t stream, uint64_t session_id) { |
|
|
|
Status GraphManager::RunGraphWithStreamAsync(const GraphId &graph_id, const std::vector<GeTensor> &inputs, |
|
|
|
std::vector<GeTensor> &outputs, rtStream_t stream, uint64_t session_id) { |
|
|
|
ErrorManager::GetInstance().SetStage(ErrorMessage::kModelCompile, ErrorMessage::kOther); |
|
|
|
std::lock_guard<std::mutex> lock(run_mutex_); |
|
|
|
GELOGI("[RunGraphWithStreamAsync] start to run graph, graph_id = %u, is_train_graph: %d", graph_id, GetTrainFlag()); |
|
|
|
@@ -1173,8 +1173,8 @@ Status RunGraphWithStreamAsync(const GraphId &graph_id, const std::vector<GeTens |
|
|
|
return ret; |
|
|
|
} |
|
|
|
|
|
|
|
auto ret = graph_executor_.ExecuteGraphWithStream(graph_id, graph_node->GetGeRootModel(), |
|
|
|
inputs, outputs, stream); |
|
|
|
ret = graph_executor_.ExecuteGraphWithStream(graph_id, graph_node->GetGeRootModel(), |
|
|
|
inputs, outputs, stream); |
|
|
|
graph_node->SetRunFlag(false); |
|
|
|
graph_node->SetIsSpecificStream(false); |
|
|
|
if (ret != SUCCESS) { |
|
|
|
|