Browse Source

!10865 Raise exception when sync stream failed

From: @jojobugfree
Reviewed-by: @kisnwang,@zhoufeng54,@chujinjin
Signed-off-by: @chujinjin
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 5 years ago
parent
commit
9591c325f7
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      mindspore/ccsrc/backend/session/ascend_session.cc
  2. +1
    -1
      mindspore/ccsrc/backend/session/gpu_session.cc

+ 1
- 1
mindspore/ccsrc/backend/session/ascend_session.cc View File

@@ -1633,7 +1633,7 @@ void AscendSession::SyncStream() {
MS_EXCEPTION_IF_NULL(runtime_instance);
auto ret = runtime_instance->SyncStream();
if (!ret) {
MS_LOG(ERROR) << "Sync stream error!";
MS_LOG(EXCEPTION) << "Sync stream error!";
}
}
} // namespace session


+ 1
- 1
mindspore/ccsrc/backend/session/gpu_session.cc View File

@@ -504,7 +504,7 @@ void GPUSession::SyncStream() {
MS_EXCEPTION_IF_NULL(runtime_instance);
auto ret = runtime_instance->SyncStream();
if (!ret) {
MS_LOG(ERROR) << "Sync stream error!";
MS_LOG(EXCEPTION) << "Sync stream error!";
}
}
} // namespace gpu


Loading…
Cancel
Save