Browse Source

bugfix: sq node is not enough for data transfer

tags/v1.2.0-rc1
zuochuanyong 4 years ago
parent
commit
664c45e23b
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/runtime/device/ascend/ascend_kernel_runtime.cc

+ 1
- 1
mindspore/ccsrc/runtime/device/ascend/ascend_kernel_runtime.cc View File

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


Loading…
Cancel
Save