blob: a6bc950a96d224a36d1a4178f5eeec53e490871d (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
|
vcpkg_from_github(
OUT_SOURCE_PATH SOURCE_PATH
REPO NVIDIA/cudnn-frontend
REF "v${VERSION}"
SHA512 75943654652bbde8dc90cf52b58078a8c8250940827cebfe96c36750bc41047d016a94f1201f9e4069cdac56e0804c2f5a534ca86d4dc0f0ba4e65a97792a7a7
HEAD_REF main
)
file(REMOVE_RECURSE "${SOURCE_PATH}/include/cudnn_frontend/thirdparty")
set(VCPKG_BUILD_TYPE release) # header only, INTERFACE library
vcpkg_find_cuda(OUT_CUDA_TOOLKIT_ROOT cuda_toolkit_root)
vcpkg_cmake_configure(
SOURCE_PATH "${SOURCE_PATH}"
OPTIONS
"-DCMAKE_CUDA_COMPILER:FILEPATH=${NVCC}"
"-DCUDAToolkit_ROOT=${cuda_toolkit_root}"
-DCUDNN_FRONTEND_BUILD_PYTHON_BINDINGS=OFF
-DCUDNN_FRONTEND_BUILD_TESTS=OFF
-DCUDNN_FRONTEND_BUILD_SAMPLES=OFF
-DCUDNN_FRONTEND_SKIP_JSON_LIB=OFF # no macro definition
MAYBE_UNUSED_VARIABLES
CUDNN_FRONTEND_FETCH_PYBINDS_IN_CMAKE
)
vcpkg_cmake_install()
vcpkg_cmake_config_fixup(CONFIG_PATH lib/cmake/cudnn_frontend PACKAGE_NAME cudnn_frontend)
# make the installed files to see nlohmann/json.hpp from vcpkg
vcpkg_replace_string(
"${CURRENT_PACKAGES_DIR}/include/cudnn_frontend_utils.h"
"\"cudnn_frontend/thirdparty/nlohmann/json.hpp\"" "<nlohmann/json.hpp>"
)
file(REMOVE_RECURSE
"${CURRENT_PACKAGES_DIR}/debug"
"${CURRENT_PACKAGES_DIR}/lib"
)
vcpkg_install_copyright(FILE_LIST "${SOURCE_PATH}/LICENSE.txt")
|