Browse Source

update deviceid

tags/v1.2.0
taoxiangdong 3 years ago
parent
commit
46179fe5e2
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/init/gelib.cc

+ 1
- 1
ge/init/gelib.cc View File

@@ -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.");


Loading…
Cancel
Save