Browse Source

Raise exception when Sync stream failed

tags/v1.2.0-rc1
caifubi 5 years ago
parent
commit
ea2aa7dec4
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

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


Loading…
Cancel
Save