[nvidia-cutlass] Fix finding cudnn (#48953)

This commit is contained in:
TLescoatTFX 2025-12-18 18:58:35 +01:00 committed by GitHub
parent 2b1b29ed11
commit 2c953609a0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
5 changed files with 38 additions and 1 deletions

View File

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

View File

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

View File

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

View File

@ -7006,7 +7006,7 @@
},
"nvidia-cutlass": {
"baseline": "4.3.1",
"port-version": 0
"port-version": 1
},
"nvtt": {
"baseline": "2.1.2",

View File

@ -1,5 +1,10 @@
{
"versions": [
{
"git-tree": "4b8b0baa5678553b337b62d27a5d63181a8ad8cf",
"version": "4.3.1",
"port-version": 1
},
{
"git-tree": "1afd6dcf96aa421e2ff8849c2fee7f220f294eb4",
"version": "4.3.1",