use non blocking cuda stream during creation

uber · Mar 30, 2020 · 76663d4 · 76663d4
1 parent f5e6dc3
commit 76663d4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cgoutils/memory/cuda_malloc.cu b/cgoutils/memory/cuda_malloc.cu
@@ -72,7 +72,7 @@ CGoCallResHandle CreateCudaStream(int device) {
   CGoCallResHandle resHandle = {NULL, NULL};
   cudaSetDevice(device);
   cudaStream_t s = NULL;
-  cudaStreamCreate(&s);
+  cudaStreamCreateWithFlags(&s, cudaStreamNonBlocking);
   resHandle.res = reinterpret_cast<void *>(s);
   resHandle.pStrErr = checkCUDAError("CreateCudaStream");
   return resHandle;