Browse Source

!2054 fix mix target with gpu target

Merge pull request !2054 from kisnwang/mix-target-check
tags/v0.5.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
c27d4157b5
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/vm/backend.cc

+ 1
- 1
mindspore/ccsrc/vm/backend.cc View File

@@ -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.";
}


Loading…
Cancel
Save