diff --git a/mindspore/ccsrc/runtime/device/ascend/ascend_kernel_runtime.cc b/mindspore/ccsrc/runtime/device/ascend/ascend_kernel_runtime.cc index e3e97a1b63..3edd307bdb 100644 --- a/mindspore/ccsrc/runtime/device/ascend/ascend_kernel_runtime.cc +++ b/mindspore/ccsrc/runtime/device/ascend/ascend_kernel_runtime.cc @@ -792,7 +792,7 @@ bool AscendKernelRuntime::InitDevice() { return false; } - ret = rtStreamCreate(&stream_, 0); + ret = rtStreamCreateWithFlags(&stream_, 0, RT_STREAM_HUGE); if (ret != RT_ERROR_NONE) { MS_LOG(EXCEPTION) << "Call rtStreamCreate, ret[" << ret << "]"; }