diff options
| author | Ethan Morgan <ethan@gweithio.com> | 2026-02-14 16:44:06 +0000 |
|---|---|---|
| committer | Ethan Morgan <ethan@gweithio.com> | 2026-02-14 16:44:06 +0000 |
| commit | 54409423f767d8b1cf30cb7d0efca6b4ca138823 (patch) | |
| tree | d915ac7828703ce4b963efdd9728a1777ba18c1e /vcpkg/ports/nvidia-cutlass/portfile.cmake | |
Diffstat (limited to 'vcpkg/ports/nvidia-cutlass/portfile.cmake')
| -rw-r--r-- | vcpkg/ports/nvidia-cutlass/portfile.cmake | 50 |
1 files changed, 50 insertions, 0 deletions
diff --git a/vcpkg/ports/nvidia-cutlass/portfile.cmake b/vcpkg/ports/nvidia-cutlass/portfile.cmake new file mode 100644 index 0000000..3991ee1 --- /dev/null +++ b/vcpkg/ports/nvidia-cutlass/portfile.cmake @@ -0,0 +1,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")
|