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