From ce78676dd4fb327153327e1d4cc0703bb7582b88 Mon Sep 17 00:00:00 2001 From: zhupuxu Date: Tue, 3 Aug 2021 09:56:49 +0800 Subject: [PATCH] fix for graph id Signed-off-by: zhupuxu --- ge/session/inner_session.cc | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/ge/session/inner_session.cc b/ge/session/inner_session.cc index b9c44ef1..eb010337 100755 --- a/ge/session/inner_session.cc +++ b/ge/session/inner_session.cc @@ -232,9 +232,6 @@ Status InnerSession::GetVariable(const std::string &name, Tensor &val) { Status InnerSession::AddGraph(uint32_t graph_id, const Graph &graph) { std::map options; - auto device_id = GetContext().DeviceId(); - GELOGD("Device id is %u", device_id); - ProfilingManager::Instance().SetGraphIdToDeviceMap(graph_id, device_id); return AddGraph(graph_id, graph, options); } @@ -248,6 +245,9 @@ Status InnerSession::AddGraph(uint32_t graph_id, const Graph &graph, session_id_, graph_id); return GE_SESS_INIT_FAILED; } + auto device_id = GetContext().DeviceId(); + GELOGD("Device id is %u", device_id); + ProfilingManager::Instance().SetGraphIdToDeviceMap(graph_id, device_id); UpdateThreadContext(options); Status ret = graph_manager_.AddGraph(graph_id, graph, options, domi::GetContext()); if (ret != SUCCESS) {