Fix Segmentation fault on Ascend

4 years ago · 7bfc2a3143
parent 9e27ca929a
commit 7bfc2a3143
1 changed files with 4 additions and 0 deletions
--- a/mindspore/ccsrc/runtime/device/ascend/ascend_kernel_runtime.cc
+++ b/mindspore/ccsrc/runtime/device/ascend/ascend_kernel_runtime.cc
@ -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;