Fix no inference detections bug when cudnn is disabled. (AlexeyAB#7617)

Fixes issue described here: AlexeyAB#7616
Tossy0423 · Apr 19, 2021 · 8bc512d · 8bc512d
1 parent b6cd759
commit 8bc512d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/dark_cuda.c b/src/dark_cuda.c
@@ -123,8 +123,11 @@ cudaStream_t get_cuda_stream() {
     int i = cuda_get_device();
     if (!streamInit[i]) {
         printf("Create CUDA-stream - %d \n", i);
-        //cudaError_t status = cudaStreamCreate(&streamsArray[i], cudaStreamNonBlocking);
+#ifdef CUDNN
         cudaError_t status = cudaStreamCreateWithFlags(&streamsArray[i], cudaStreamNonBlocking);
+#else
+        cudaError_t status = cudaStreamCreate(&streamsArray[i]);
+#endif
         if (status != cudaSuccess) {
             printf(" cudaStreamCreate error: %d \n", status);
             const char *s = cudaGetErrorString(status);