Browse Source

fix for graph id

Signed-off-by: zhupuxu <zhupuxu@huawei.com>
pull/2070/head
zhupuxu 3 years ago
parent
commit
ce78676dd4
1 changed files with 3 additions and 3 deletions
  1. +3
    -3
      ge/session/inner_session.cc

+ 3
- 3
ge/session/inner_session.cc View File

@@ -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<std::string, std::string> 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) {


Loading…
Cancel
Save