From 260126dd64852bb8416fee4503356bba283aadee Mon Sep 17 00:00:00 2001
From: weiyang <yangwei79@huawei.com>
Date: Thu, 28 Jan 2021 10:50:47 +0800
Subject: [PATCH] session device id

---
 ge/session/inner_session.cc | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/ge/session/inner_session.cc b/ge/session/inner_session.cc
index c4f8a53b..92d09b26 100755
--- a/ge/session/inner_session.cc
+++ b/ge/session/inner_session.cc
@@ -77,6 +77,23 @@ Status InnerSession::Initialize() {
 
   UpdateThreadContext(std::map<std::string, std::string>{});
 
+  // session device id set here
+  std::string str_session_device_id;
+  if (GetContext().GetOption("ge.session_device_id", str_session_device_id) == SUCCESS) {
+    GELOGI("Option session device id has set, value is %s.", str_session_device_id.c_str());
+
+    uint32_t session_device_id = 0;
+    try {
+      session_device_id = static_cast<uint32_t>(std::stoi(str_session_device_id.c_str()));
+      // session device id has priority
+      GetContext().SetCtxDeviceId(session_device_id);
+    } catch (std::invalid_argument &) {
+      GELOGW("session device id %s transform to int failed.", str_session_device_id.c_str());
+    } catch (std::out_of_range &) {
+      GELOGW("session device id %s transform to int failed.", str_session_device_id.c_str());
+    }
+  }
+
   GE_CHK_RT_RET(rtSetDevice(GetContext().DeviceId()));
 
   DumpProperties dump_properties;