From 300406e26a155d8cf35f90985c478b9cea82d264 Mon Sep 17 00:00:00 2001
From: cyy <cyyever@outlook.com>
Date: Fri, 10 Jan 2020 17:47:10 +0800
Subject: [PATCH] fix macro

---
 src/dark_cuda.c | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/dark_cuda.c b/src/dark_cuda.c
index cfb6b825..d16f9efc 100644
--- a/src/dark_cuda.c
+++ b/src/dark_cuda.c
@@ -162,7 +162,7 @@ cudnnHandle_t cudnn_handle()
         cudnnCreate(&handle[i]);
         init[i] = 1;
         cudnnStatus_t status = cudnnSetStream(handle[i], get_cuda_stream());
-        CHECK_CUDA(status);
+        CHECK_CUDNN(status);
     }
     return handle[i];
 }