OpenI
/
MegEngine
Not watched
1
0
Fork 0
Code
Releases 31 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
Browse Source

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

GitOrigin-RevId: 49d8842dd4
tags/v1.0.0-rc1
Megvii Engine Team 4 years ago
parent
14235385e2
commit
bd73dabbe2
1 changed files with 1 additions and 0 deletions
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +1
    -0
      dnn/test/CMakeLists.txt

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

@@ -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


Write Preview
Loading…
Cancel
Save