From bd73dabbe279cdc6cb44aaa868d700034abfc500 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Wed, 2 Sep 2020 14:50:38 +0800
Subject: [PATCH] fix(dnn/build): add CUDNN_INCLUDE_DIR to the megdnn_test
 target

GitOrigin-RevId: 49d8842dd48ec754a60db7328d52895ed70bb6b0
---
 dnn/test/CMakeLists.txt | 1 +
 1 file changed, 1 insertion(+)

diff --git a/dnn/test/CMakeLists.txt b/dnn/test/CMakeLists.txt
index 5d24a613..1faa5e3a 100644
--- a/dnn/test/CMakeLists.txt
+++ b/dnn/test/CMakeLists.txt
@@ -49,6 +49,7 @@ target_link_libraries(megdnn_test megdnn ${MGE_BLAS_LIBS})
 
 if (MGE_WITH_CUDA)
     target_link_libraries(megdnn_test cutlass)
+    target_include_directories(megdnn_test PRIVATE ${CUDNN_INCLUDE_DIR})
 endif()
 
 target_include_directories(megdnn_test