fix mix target device id

5 years ago · 04d01a36f4
parent 65189e8ccc
commit 04d01a36f4
1 changed files with 4 additions and 1 deletions
--- 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;
 }