From d290c07a0fbb7a8f3a3a26af81f1dc14e107f659 Mon Sep 17 00:00:00 2001
From: kswang <wangkaisheng2@huawei.com>
Date: Tue, 30 Jun 2020 15:19:25 +0800
Subject: [PATCH] fix mix target device id

---
 mindspore/ccsrc/vm/backend.cc | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/mindspore/ccsrc/vm/backend.cc b/mindspore/ccsrc/vm/backend.cc
index 3fde263c9..4341427a6 100644
--- a/mindspore/ccsrc/vm/backend.cc
+++ b/mindspore/ccsrc/vm/backend.cc
@@ -353,7 +353,10 @@ void MsBackend::CreateOtherSession(const std::string &target) {
   if (other_sess_ == nullptr) {
     MS_LOG(EXCEPTION) << "Session create failed!, please make sure target device:" << target << " is available.";
   }
-  other_sess_->Init(0);
+  auto context_ptr = MsContext::GetInstance();
+  MS_EXCEPTION_IF_NULL(context_ptr);
+  uint32_t device_id = context_ptr->device_id();
+  other_sess_->Init(device_id);
   other_sess_->RegisterSummaryCallBackFunc(callbacks::SummarySaveCallback);
   other_device_ = target;
 }
-- 
GitLab