fix(dnn/build): add CUDNN_INCLUDE_DIR to the megdnn_test target

GitOrigin-RevId: 49d8842dd48ec754a60db7328d52895ed70bb6b0

fix(dnn/build): add CUDNN_INCLUDE_DIR to the megdnn_test target
GitOrigin-RevId: 49d8842dd48ec754a60db7328d52895ed70bb6b0
bd73dabb · Megvii Engine Team · 14235385 · bd73dabb
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

dnn/test/CMakeLists.txt dnn/test/CMakeLists.txt +1 -0

未找到文件。
--- 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