bugfix: sq node is not enough for data transfer

pull/13058/head
zuochuanyong 4 years ago
parent 0a06e0da17
commit 664c45e23b

@ -792,7 +792,7 @@ bool AscendKernelRuntime::InitDevice() {
return false; return false;
} }
ret = rtStreamCreate(&stream_, 0); ret = rtStreamCreateWithFlags(&stream_, 0, RT_STREAM_HUGE);
if (ret != RT_ERROR_NONE) { if (ret != RT_ERROR_NONE) {
MS_LOG(EXCEPTION) << "Call rtStreamCreate, ret[" << ret << "]"; MS_LOG(EXCEPTION) << "Call rtStreamCreate, ret[" << ret << "]";
} }

Loading…
Cancel
Save