From 664c45e23b9e5ae6f481e0b3b565af02fac2909e Mon Sep 17 00:00:00 2001
From: zuochuanyong <zuochuanyong@huawei.com>
Date: Tue, 9 Mar 2021 06:40:44 -0500
Subject: [PATCH] bugfix: sq node is not enough for data transfer

---
 mindspore/ccsrc/runtime/device/ascend/ascend_kernel_runtime.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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 << "]";
   }