From 76663d463c58316b03c454c4d7364ab604fe6e2b Mon Sep 17 00:00:00 2001
From: Jian Shen <jians@uber.com>
Date: Mon, 2 Mar 2020 11:24:19 -0800
Subject: [PATCH] use non blocking cuda stream during creation

---
 cgoutils/memory/cuda_malloc.cu | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cgoutils/memory/cuda_malloc.cu b/cgoutils/memory/cuda_malloc.cu
index b826f3c6..6c49eee0 100644
--- 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;