mirror of https://github.com/microsoft/vcpkg.git
[nvidia-cutlass] Fix finding cudnn (#48953)
This commit is contained in:
parent
2b1b29ed11
commit
2c953609a0
|
|
@ -0,0 +1,24 @@
|
|||
diff --git a/cuDNN.cmake b/cuDNN.cmake
|
||||
index 30b58581..fdbc8314 100644
|
||||
--- a/cuDNN.cmake
|
||||
+++ b/cuDNN.cmake
|
||||
@@ -43,7 +43,8 @@ find_path(
|
||||
$ENV{CUDNN_PATH}/include
|
||||
$ENV{CUDA_PATH}/include
|
||||
${CUDNN_PATH}/include
|
||||
- /usr/include)
|
||||
+ /usr/include
|
||||
+ $ENV{CUDNN_PATH})
|
||||
|
||||
find_library(
|
||||
_CUDNN_LIBRARY cudnn
|
||||
@@ -61,7 +62,8 @@ find_library(
|
||||
${CUDNN_PATH}/lib/x64
|
||||
${CUDNN_PATH}/lib
|
||||
/usr/lib/x86_64-linux-gnu
|
||||
- /usr/lib)
|
||||
+ /usr/lib
|
||||
+ $ENV{CUDNN_PATH})
|
||||
|
||||
if(_CUDNN_INCLUDE_DIR AND _CUDNN_LIBRARY)
|
||||
|
||||
|
|
@ -4,6 +4,8 @@ vcpkg_from_github(
|
|||
REF "v${VERSION}"
|
||||
SHA512 16ef4f68d819f50f4e3f250ddf41fc19105fa1155a86fc42970be0f84e32129cfb05f2730466e9491e84ca5faed595b3e5fbeee7c48a27da812e1927f842623e
|
||||
HEAD_REF main
|
||||
PATCHES
|
||||
fix-cudnn-path.patch
|
||||
)
|
||||
|
||||
vcpkg_find_acquire_program(PYTHON3)
|
||||
|
|
@ -16,6 +18,11 @@ list(APPEND FEATURE_OPTIONS
|
|||
"-DCUDAToolkit_ROOT=${cuda_toolkit_root}"
|
||||
)
|
||||
|
||||
list(APPEND CMAKE_MODULE_PATH "${CURRENT_INSTALLED_DIR}/share/cudnn")
|
||||
find_package(CUDNN REQUIRED)
|
||||
get_filename_component(CUDNN_LIBRARY_DIR "${CUDNN_LIBRARIES}" DIRECTORY)
|
||||
set(ENV{CUDNN_PATH} "${CUDNN_LIBRARY_DIR};${CUDNN_INCLUDE_DIRS}")
|
||||
|
||||
|
||||
vcpkg_cmake_configure(
|
||||
SOURCE_PATH "${SOURCE_PATH}"
|
||||
|
|
|
|||
|
|
@ -1,6 +1,7 @@
|
|||
{
|
||||
"name": "nvidia-cutlass",
|
||||
"version": "4.3.1",
|
||||
"port-version": 1,
|
||||
"description": "CUDA Templates for Linear Algebra Subroutines",
|
||||
"homepage": "https://github.com/NVIDIA/cutlass",
|
||||
"license": "BSD-3-Clause",
|
||||
|
|
|
|||
|
|
@ -7006,7 +7006,7 @@
|
|||
},
|
||||
"nvidia-cutlass": {
|
||||
"baseline": "4.3.1",
|
||||
"port-version": 0
|
||||
"port-version": 1
|
||||
},
|
||||
"nvtt": {
|
||||
"baseline": "2.1.2",
|
||||
|
|
|
|||
|
|
@ -1,5 +1,10 @@
|
|||
{
|
||||
"versions": [
|
||||
{
|
||||
"git-tree": "4b8b0baa5678553b337b62d27a5d63181a8ad8cf",
|
||||
"version": "4.3.1",
|
||||
"port-version": 1
|
||||
},
|
||||
{
|
||||
"git-tree": "1afd6dcf96aa421e2ff8849c2fee7f220f294eb4",
|
||||
"version": "4.3.1",
|
||||
|
|
|
|||
Loading…
Reference in New Issue