Skip to content

Commit 4c802cc

Browse files
authored
Fix cudnn.make include path (#73005)
* Fix cudnn.make * Fix
1 parent e9ad101 commit 4c802cc

File tree

1 file changed

+11
-7
lines changed

1 file changed

+11
-7
lines changed

cmake/cudnn.cmake

Lines changed: 11 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -10,20 +10,24 @@ else()
1010
CACHE PATH "CUDNN ROOT")
1111
endif()
1212

13+
set(TARGET_ARCH "x86_64")
14+
if(NOT ${CMAKE_SYSTEM_PROCESSOR})
15+
set(TARGET_ARCH ${CMAKE_SYSTEM_PROCESSOR})
16+
endif()
17+
1318
find_path(
1419
CUDNN_INCLUDE_DIR cudnn.h
15-
PATHS ${CUDNN_ROOT} ${CUDNN_ROOT}/include $ENV{CUDNN_ROOT}
16-
$ENV{CUDNN_ROOT}/include ${CUDA_TOOLKIT_INCLUDE}
20+
PATHS ${CUDNN_ROOT}
21+
${CUDNN_ROOT}/include
22+
${CUDNN_ROOT}/include/${TARGET_ARCH}-linux-gnu
23+
$ENV{CUDNN_ROOT}
24+
$ENV{CUDNN_ROOT}/include
25+
${CUDA_TOOLKIT_INCLUDE}
1726
/usr/local/lib/python${PY_VERSION}/dist-packages/nvidia/cudnn/include/
1827
NO_DEFAULT_PATH)
1928

2029
get_filename_component(__libpath_hist ${CUDA_CUDART_LIBRARY} PATH)
2130

22-
set(TARGET_ARCH "x86_64")
23-
if(NOT ${CMAKE_SYSTEM_PROCESSOR})
24-
set(TARGET_ARCH ${CMAKE_SYSTEM_PROCESSOR})
25-
endif()
26-
2731
list(
2832
APPEND
2933
CUDNN_CHECK_LIBRARY_DIRS

0 commit comments

Comments
 (0)