|
|
|
@ -293,7 +293,7 @@ Status GraphManager::InitDynamicParams(ComputeGraphPtr &compute_graph) {
|
|
|
|
|
return FAILED;
|
|
|
|
|
}
|
|
|
|
|
if ((op_desc->GetType() == DATA) || (op_type == kGetNextName)) {
|
|
|
|
|
GELOGI("Need to process multi batch for compute graph.");
|
|
|
|
|
GELOGI("Need to process multi batch for compute graph. op_type:%s", op_desc->GetType(),c_str());
|
|
|
|
|
GetLocalOmgContext().need_multi_batch = true;
|
|
|
|
|
break;
|
|
|
|
|
}
|
|
|
|
@ -348,7 +348,7 @@ Status GraphManager::AddGraph(const GraphId &graph_id, const Graph &graph,
|
|
|
|
|
for (auto &subgraph : compute_graph->GetAllSubgraphs()) {
|
|
|
|
|
(void)AttrUtils::SetStr(*subgraph, ATTR_NAME_SESSION_GRAPH_ID, session_graph_id);
|
|
|
|
|
}
|
|
|
|
|
GELOGD("Get graph session_graph_id attr failed, set session id to default value: [0]");
|
|
|
|
|
GELOGD("Get graph session_graph_id attr failed, set session id to default value: [0].");
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
GraphNodePtr graph_node = MakeShared<ge::GraphNode>(graph_id);
|
|
|
|
@ -734,8 +734,8 @@ Status GraphManager::PreRunAfterOptimizeSubGraph(const GraphNodePtr &graph_node,
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
Status GraphManager::SetRtContext(rtContext_t rt_context, rtCtxMode_t mode, uint64_t session_id, uint32_t graph_id) {
|
|
|
|
|
GELOGD("set rt_context, session id: %lu, graph id: %u, mode %d, device id:%u.", session_id, graph_id,
|
|
|
|
|
static_cast<int>(mode), ge::GetContext().DeviceId());
|
|
|
|
|
GELOGD("set rt_context, session id: %lu, graph id: %u, mode %d, device id:%u.",
|
|
|
|
|
session_id, graph_id, static_cast<int>(mode), ge::GetContext().DeviceId());
|
|
|
|
|
|
|
|
|
|
rtError_t rt_ret = rtCtxCreate(&rt_context, mode, ge::GetContext().DeviceId());
|
|
|
|
|
if (rt_ret != RT_ERROR_NONE) {
|
|
|
|
@ -758,7 +758,7 @@ Status GraphManager::RunCustomPass(const GraphNodePtr &graph_node) {
|
|
|
|
|
|
|
|
|
|
GE_TIMESTAMP_START(RunCustomPass);
|
|
|
|
|
GraphPtr graph = std::const_pointer_cast<Graph>(const_graph);
|
|
|
|
|
GE_CHK_STATUS_RET(CustomPassHelper::Instance().Run(graph), "Graph[%s] run custom pass fail.",
|
|
|
|
|
GE_CHK_STATUS_RET(CustomPassHelper::Instance().Run(graph), "Graph[%s] run custom pass fail",
|
|
|
|
|
comp_graph->GetName().c_str());
|
|
|
|
|
GE_TIMESTAMP_END(RunCustomPass, "GraphBuilder::RunCustomPass");
|
|
|
|
|
return SUCCESS;
|
|
|
|
@ -776,7 +776,7 @@ Status GraphManager::PreRun(const GraphNodePtr &graph_node, const std::vector<Ge
|
|
|
|
|
GE_CHK_STATUS_RET(analyzer_instance->BuildJsonObject(session_id, compute_graph->GetGraphID()),
|
|
|
|
|
"BuildJsonObject Failed")
|
|
|
|
|
|
|
|
|
|
GEEVENT("PreRun start, graph node size %zu, session id %lu, graph id %u, graph name %s",
|
|
|
|
|
GEEVENT("PreRun start, graph node size %zu, session id %lu, graph id %u, graph name %s.",
|
|
|
|
|
compute_graph->GetDirectNodesSize(), session_id, compute_graph->GetGraphID(),
|
|
|
|
|
compute_graph->GetName().c_str());
|
|
|
|
|
GE_DUMP(compute_graph, "PreRunBegin");
|
|
|
|
@ -797,7 +797,7 @@ Status GraphManager::PreRun(const GraphNodePtr &graph_node, const std::vector<Ge
|
|
|
|
|
if (run_optimize_original_graph) {
|
|
|
|
|
Status ret = PreRunOptimizeOriginalGraph(graph_node, inputs, compute_graph, session_id);
|
|
|
|
|
if (ret != SUCCESS) {
|
|
|
|
|
GELOGE(ret, "Run PreRunOptimizeOriginalGraph failed for graph:%s.", compute_graph->GetName().c_str());
|
|
|
|
|
GELOGE(ret, "Run PreRunOptimizeOriginalGraph failed for graph:%s", compute_graph->GetName().c_str());
|
|
|
|
|
return ret;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
@ -869,7 +869,7 @@ Status GraphManager::StartForRunGraph(const GraphNodePtr &graph_node, const std:
|
|
|
|
|
// release rts generate context
|
|
|
|
|
RtContextUtil::GetInstance().DestroyRtContexts(session_id, graph_node->GetGraphId());
|
|
|
|
|
if (ret != SUCCESS) {
|
|
|
|
|
GELOGE(ret, "PreRun Failed.");
|
|
|
|
|
GELOGE(ret, "PreRun Failed. graph_id:%u", graph_node->GetGraphId());
|
|
|
|
|
return ret;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
@ -1209,7 +1209,7 @@ Status GraphManager::BuildGraphForUnregisteredOp(const GraphId &graph_id, const
|
|
|
|
|
|
|
|
|
|
Status GraphManager::BuildGraph(const GraphId &graph_id, const std::vector<GeTensor> &inputs,
|
|
|
|
|
GeRootModelPtr &ge_root_model, uint64_t session_id, bool async) {
|
|
|
|
|
GELOGD("[BuildGraph] start to build graph, graph_id=%u.", graph_id);
|
|
|
|
|
GELOGD("[BuildGraph] start to build graph, graph_id:%u.", graph_id);
|
|
|
|
|
if (inputs.empty()) {
|
|
|
|
|
GELOGW("[BuildGraph] BuildGraph warning: empty GeTensor inputs");
|
|
|
|
|
}
|
|
|
|
@ -1241,7 +1241,7 @@ Status GraphManager::BuildGraph(const GraphId &graph_id, const std::vector<GeTen
|
|
|
|
|
ret = StartForRunGraph(graph_node, inputs, ge_root_model, session_id);
|
|
|
|
|
graph_node->SetRunFlag(false);
|
|
|
|
|
if (ret != SUCCESS) {
|
|
|
|
|
GELOGE(GE_GRAPH_PRERUN_FAILED, "[BuildGraph] StartForRunGraph failed!");
|
|
|
|
|
GELOGE(GE_GRAPH_PRERUN_FAILED, "[BuildGraph] StartForRunGraph failed! graph_id:%u", graph_id);
|
|
|
|
|
return GE_GRAPH_PRERUN_FAILED;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|