From 140bc1882dd48dfff93e94dae0fb780351513c9e Mon Sep 17 00:00:00 2001
From: Alexey <AlexeyAB@users.noreply.github.com>
Date: Wed, 15 Aug 2018 17:31:31 +0000
Subject: [PATCH] Merge pull request #1416 from mhaghighat/master

---
 src/darknet.c |    1 +
 1 files changed, 1 insertions(+), 0 deletions(-)

diff --git a/src/darknet.c b/src/darknet.c
index 068c9e0..1dc073b 100644
--- a/src/darknet.c
+++ b/src/darknet.c
@@ -377,6 +377,7 @@
 #else
     if(gpu_index >= 0){
         cuda_set_device(gpu_index);
+        check_error(cudaSetDeviceFlags(cudaDeviceScheduleBlockingSync));
     }
 #endif
 

--
Gitblit v1.10.0