From 46179fe5e2721ca30a8ea8a6e787d7f3c346e043 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.");