From ea2aa7dec47c21d58f1c59f6fbfd189c6e1177a0 Mon Sep 17 00:00:00 2001 From: caifubi Date: Thu, 31 Dec 2020 10:26:06 +0800 Subject: [PATCH] Raise exception when Sync stream failed --- mindspore/ccsrc/backend/session/ascend_session.cc | 2 +- mindspore/ccsrc/backend/session/gpu_session.cc | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/mindspore/ccsrc/backend/session/ascend_session.cc b/mindspore/ccsrc/backend/session/ascend_session.cc index 8f43ce27cd..4ddd88a2fb 100644 --- a/mindspore/ccsrc/backend/session/ascend_session.cc +++ b/mindspore/ccsrc/backend/session/ascend_session.cc @@ -1617,7 +1617,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 diff --git a/mindspore/ccsrc/backend/session/gpu_session.cc b/mindspore/ccsrc/backend/session/gpu_session.cc index ba946aa876..b814a2adbf 100644 --- a/mindspore/ccsrc/backend/session/gpu_session.cc +++ b/mindspore/ccsrc/backend/session/gpu_session.cc @@ -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