blob: 3991ee1d9a70040c3ba11340f0f7909ead7384fd (
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
40
41
42
43
44
45
46
47
48
49
50
|
vcpkg_from_github(
OUT_SOURCE_PATH SOURCE_PATH
REPO NVIDIA/cutlass
REF "v${VERSION}"
SHA512 a08aac281fb3bdea82c0a044dc643c40e4803d02e55bbea450021cb7a5472aed86e79c5df41cd981976af8403f18cc48d8069045c4e68339430d3a3caeb109ac
HEAD_REF main
)
vcpkg_find_acquire_program(PYTHON3)
get_filename_component(PYTHON_PATH "${PYTHON3}" PATH)
vcpkg_add_to_path(PREPEND "${PYTHON_PATH}")
vcpkg_find_cuda(OUT_CUDA_TOOLKIT_ROOT cuda_toolkit_root)
list(APPEND FEATURE_OPTIONS
"-DCMAKE_CUDA_COMPILER=${NVCC}"
"-DCUDAToolkit_ROOT=${cuda_toolkit_root}"
)
vcpkg_cmake_configure(
SOURCE_PATH "${SOURCE_PATH}"
OPTIONS
-DCMAKE_SUPPRESS_REGENERATION=ON # for some reason it keeps regenerating in Windows
"-DCUTLASS_REVISION:STRING=v${VERSION}"
-DCUTLASS_NATIVE_CUDA=OFF
-DCUTLASS_ENABLE_HEADERS_ONLY=ON
-DCUTLASS_ENABLE_TOOLS=ON
-DCUTLASS_ENABLE_LIBRARY=OFF
-DCUTLASS_ENABLE_PROFILER=OFF
-DCUTLASS_ENABLE_PERFORMANCE=OFF
-DCUTLASS_ENABLE_TESTS=OFF
-DCUTLASS_ENABLE_GTEST_UNIT_TESTS=OFF
-DCUTLASS_ENABLE_CUBLAS=ON
-DCUTLASS_ENABLE_CUDNN=ON
"-DPython3_EXECUTABLE:FILEPATH=${PYTHON3}"
${FEATURE_OPTIONS}
MAYBE_UNUSED_VARIABLES
CUTLASS_NATIVE_CUDA
)
vcpkg_cmake_install()
vcpkg_cmake_config_fixup(CONFIG_PATH "lib/cmake/NvidiaCutlass" PACKAGE_NAME "NvidiaCutlass")
file(REMOVE_RECURSE
"${CURRENT_PACKAGES_DIR}/debug"
"${CURRENT_PACKAGES_DIR}/test"
"${CURRENT_PACKAGES_DIR}/lib"
)
vcpkg_install_copyright(FILE_LIST "${SOURCE_PATH}/LICENSE.txt")
|
