diff --git a/mindspore/ccsrc/vm/backend.cc b/mindspore/ccsrc/vm/backend.cc index fe84bfbff39b577f2119dbc33d9da96d87baecca..3fde263c9db3a7e90c7429110861f0893eb1983a 100644 --- a/mindspore/ccsrc/vm/backend.cc +++ b/mindspore/ccsrc/vm/backend.cc @@ -349,7 +349,7 @@ void MsBackend::CreateOtherSession(const std::string &target) { if (other_sess_ != nullptr && other_device_ == target) { return; } - other_sess_ = session::SessionFactory::Get().Create(kCPUDevice); + other_sess_ = session::SessionFactory::Get().Create(target); if (other_sess_ == nullptr) { MS_LOG(EXCEPTION) << "Session create failed!, please make sure target device:" << target << " is available."; }