From 4a13c1edfb2d0cb7138ec2b8c8f6d1e76930c29e Mon Sep 17 00:00:00 2001 From: taoxiangdong Date: Fri, 11 Dec 2020 14:12:13 +0800 Subject: [PATCH] update deviceid --- ge/init/gelib.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ge/init/gelib.cc b/ge/init/gelib.cc index 20859c07..03028e53 100755 --- a/ge/init/gelib.cc +++ b/ge/init/gelib.cc @@ -146,7 +146,7 @@ Status GELib::InnerInitialize(const map &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.");