DEBUG=1 fixed

6 years ago · ce2e0eff00
parent d767e8ca38
commit ce2e0eff00
2 changed files with 3 additions and 2 deletions
--- a/3
+++ b/3
@ -60,7 +60,8 @@ COMMON= -Iinclude/
 CFLAGS=-Wall -Wfatal-errors -Wno-unused-result -Wno-unknown-pragmas -fPIC
 ifeq ($(DEBUG), 1) 
-OPTS= -O0 -g
+#OPTS= -O0 -g
 #OPTS= -Og -g
 COMMON+= -DDEBUG
 CFLAGS+= -DDEBUG
 else
--- a/src/cuda.c
+++ b/src/cuda.c
@ -13,7 +13,7 @@ void cuda_set_device(int n)
 {
    gpu_index = n;
    cudaError_t status = cudaSetDevice(n);
-    CHECK_CUDA(status);
+    if(status != cudaSuccess) CHECK_CUDA(status);
 }
 int cuda_get_device()