p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!10838 Fix Segmentation fault on Ascend 

From: @jojobugfree
Reviewed-by: @kisnwang,@chujinjin
Signed-off-by: @kisnwang
This commit is contained in:
mindspore-ci-bot 2020-12-30 20:11:57 +08:00 committed by Gitee
parent 9b2d79c14c 7bfc2a3143
commit b1ca1fbdb9
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/ccsrc/runtime/device/ascend/ascend_kernel_runtime.cc
View File

@ -710,6 +710,10 @@ bool AscendKernelRuntime::RunTask(const session::KernelGraph *graph) {
bool AscendKernelRuntime::SyncStream() {
InnerSetContext();
if (stream_ == nullptr) {
MS_LOG(ERROR) << "SyncStream failed. stream_ is nullptr";
return false;
}
if (RT_ERROR_NONE != rtStreamSynchronize(stream_)) { // o for switch stream
MS_LOG(ERROR) << "Call runtime rtStreamSynchronize error.";
return false;