update deviceid

4 years ago · 46179fe5e2
parent c3be30b9ce
commit 46179fe5e2
1 changed files with 1 additions and 1 deletions
--- a/ge/init/gelib.cc
+++ b/ge/init/gelib.cc
@ -146,7 +146,7 @@ Status GELib::InnerInitialize(const map<string, string> &options) {
  }

  if (is_train_mode_ || (options_.device_id != kDefaultDeviceIdForInfer)) {
-    GE_CHK_RT_RET(rtSetDevice(options.device_id));
+    GE_CHK_RT_RET(rtSetDevice(options_.device_id));
  }

  GELOGI("sessionManager initial.");