|
| 1 | +cmake_minimum_required(VERSION 3.18) |
| 2 | + |
| 3 | +# Default GPU architectures. Users can override with |
| 4 | +# -DCMAKE_CUDA_ARCHITECTURES=... ; sm_100 (Blackwell) is appended |
| 5 | +# automatically when the CUDA toolkit is new enough to know it. |
| 6 | +if(NOT DEFINED CMAKE_CUDA_ARCHITECTURES) |
| 7 | + set(CMAKE_CUDA_ARCHITECTURES 80 90) |
| 8 | + set(_fsz_default_archs ON) |
| 9 | +endif() |
| 10 | + |
| 11 | +project(fsz LANGUAGES CXX CUDA VERSION 1.0.0) |
| 12 | + |
| 13 | +if(_fsz_default_archs AND CMAKE_CUDA_COMPILER_VERSION VERSION_GREATER_EQUAL 12.8) |
| 14 | + list(APPEND CMAKE_CUDA_ARCHITECTURES 100) |
| 15 | +endif() |
| 16 | + |
| 17 | +set(CMAKE_CXX_STANDARD 17) |
| 18 | +set(CMAKE_CXX_STANDARD_REQUIRED ON) |
| 19 | +set(CMAKE_CUDA_STANDARD 17) |
| 20 | +set(CMAKE_CUDA_STANDARD_REQUIRED ON) |
| 21 | + |
| 22 | +if(NOT CMAKE_BUILD_TYPE) |
| 23 | + set(CMAKE_BUILD_TYPE Release CACHE STRING "" FORCE) |
| 24 | +endif() |
| 25 | + |
| 26 | +find_package(CUDAToolkit REQUIRED) |
| 27 | + |
| 28 | +option(FSZ_BUILD_SHARED "Also build the shared library (libfsz.so)" ON) |
| 29 | +option(FSZ_BUILD_TOOLS "Build the fsz command-line tool" ON) |
| 30 | +option(FSZ_BUILD_EXAMPLES "Build FSZ examples" ON) |
| 31 | +option(FSZ_BUILD_TESTS "Build FSZ tests" ON) |
| 32 | + |
| 33 | +set(FSZ_SOURCES src/fsz.cu) |
| 34 | + |
| 35 | +# Static library (the default link target, exported as fsz::fsz). |
| 36 | +add_library(fsz STATIC ${FSZ_SOURCES}) |
| 37 | +add_library(fsz::fsz ALIAS fsz) |
| 38 | + |
| 39 | +# Shared library for dynamic linking and the Python bindings. |
| 40 | +if(FSZ_BUILD_SHARED) |
| 41 | + add_library(fsz_shared SHARED ${FSZ_SOURCES}) |
| 42 | + set_target_properties(fsz_shared PROPERTIES OUTPUT_NAME fsz) |
| 43 | +endif() |
| 44 | + |
| 45 | +foreach(_fsz_lib IN ITEMS fsz fsz_shared) |
| 46 | + if(NOT TARGET ${_fsz_lib}) |
| 47 | + continue() |
| 48 | + endif() |
| 49 | + target_include_directories(${_fsz_lib} |
| 50 | + PUBLIC $<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}/include> |
| 51 | + $<INSTALL_INTERFACE:include> |
| 52 | + PRIVATE ${CMAKE_CURRENT_SOURCE_DIR}/src) |
| 53 | + target_link_libraries(${_fsz_lib} PUBLIC CUDA::cudart) |
| 54 | + set_target_properties(${_fsz_lib} PROPERTIES |
| 55 | + CUDA_SEPARABLE_COMPILATION OFF |
| 56 | + POSITION_INDEPENDENT_CODE ON) |
| 57 | + target_compile_options(${_fsz_lib} PRIVATE |
| 58 | + $<$<COMPILE_LANGUAGE:CUDA>:-O3>) |
| 59 | +endforeach() |
| 60 | + |
| 61 | +# Command-line tool: the binary is named `fsz`. |
| 62 | +if(FSZ_BUILD_TOOLS) |
| 63 | + add_executable(fsz_cli tools/fsz.cu) |
| 64 | + set_target_properties(fsz_cli PROPERTIES OUTPUT_NAME fsz) |
| 65 | + target_include_directories(fsz_cli PRIVATE tools) |
| 66 | + target_link_libraries(fsz_cli PRIVATE fsz::fsz) |
| 67 | +endif() |
| 68 | + |
| 69 | +if(FSZ_BUILD_EXAMPLES) |
| 70 | + add_executable(fsz_example_roundtrip examples/example_roundtrip.cu) |
| 71 | + target_link_libraries(fsz_example_roundtrip PRIVATE fsz::fsz) |
| 72 | + |
| 73 | + add_executable(fsz_example_hostptr examples/example_hostptr.cu) |
| 74 | + target_link_libraries(fsz_example_hostptr PRIVATE fsz::fsz) |
| 75 | +endif() |
| 76 | + |
| 77 | +if(FSZ_BUILD_TESTS) |
| 78 | + enable_testing() |
| 79 | + add_executable(fsz_test_roundtrip tests/test_roundtrip.cu) |
| 80 | + target_link_libraries(fsz_test_roundtrip PRIVATE fsz::fsz) |
| 81 | + add_test(NAME roundtrip COMMAND fsz_test_roundtrip) |
| 82 | + |
| 83 | + add_executable(fsz_test_hostptr tests/test_hostptr.cu) |
| 84 | + target_include_directories(fsz_test_hostptr PRIVATE src) |
| 85 | + target_link_libraries(fsz_test_hostptr PRIVATE fsz::fsz) |
| 86 | + add_test(NAME hostptr COMMAND fsz_test_hostptr) |
| 87 | +endif() |
| 88 | + |
| 89 | +include(GNUInstallDirs) |
| 90 | +include(CMakePackageConfigHelpers) |
| 91 | + |
| 92 | +# Fortran interface: built whenever a Fortran compiler is available. |
| 93 | +option(FSZ_BUILD_FORTRAN "Build the Fortran interface" ON) |
| 94 | +if(FSZ_BUILD_FORTRAN) |
| 95 | + include(CheckLanguage) |
| 96 | + check_language(Fortran) |
| 97 | + if(CMAKE_Fortran_COMPILER) |
| 98 | + enable_language(Fortran) |
| 99 | + add_library(fsz_fortran STATIC fortran/fsz.f90) |
| 100 | + target_link_libraries(fsz_fortran PUBLIC fsz::fsz stdc++) |
| 101 | + set_target_properties(fsz_fortran PROPERTIES |
| 102 | + Fortran_MODULE_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}/fortran_modules) |
| 103 | + target_include_directories(fsz_fortran INTERFACE |
| 104 | + $<BUILD_INTERFACE:${CMAKE_CURRENT_BINARY_DIR}/fortran_modules>) |
| 105 | + if(FSZ_BUILD_EXAMPLES) |
| 106 | + add_executable(fsz_example_fortran fortran/example_fsz.f90) |
| 107 | + target_link_libraries(fsz_example_fortran PRIVATE fsz_fortran) |
| 108 | + endif() |
| 109 | + if(FSZ_BUILD_TESTS) |
| 110 | + add_executable(fsz_test_fortran fortran/test_fsz.f90) |
| 111 | + target_link_libraries(fsz_test_fortran PRIVATE fsz_fortran) |
| 112 | + add_test(NAME fortran COMMAND fsz_test_fortran) |
| 113 | + endif() |
| 114 | + install(TARGETS fsz_fortran |
| 115 | + ARCHIVE DESTINATION ${CMAKE_INSTALL_LIBDIR}) |
| 116 | + install(FILES ${CMAKE_CURRENT_BINARY_DIR}/fortran_modules/fsz.mod |
| 117 | + DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}/fsz) |
| 118 | + else() |
| 119 | + message(STATUS "FSZ: no Fortran compiler found; skipping the Fortran interface") |
| 120 | + endif() |
| 121 | +endif() |
| 122 | + |
| 123 | +install(TARGETS fsz EXPORT fszTargets |
| 124 | + ARCHIVE DESTINATION ${CMAKE_INSTALL_LIBDIR} |
| 125 | + LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR}) |
| 126 | +if(FSZ_BUILD_SHARED) |
| 127 | + install(TARGETS fsz_shared EXPORT fszTargets |
| 128 | + LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR}) |
| 129 | +endif() |
| 130 | +if(FSZ_BUILD_TOOLS) |
| 131 | + install(TARGETS fsz_cli RUNTIME DESTINATION ${CMAKE_INSTALL_BINDIR}) |
| 132 | +endif() |
| 133 | +install(DIRECTORY include/ DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}) |
| 134 | + |
| 135 | +install(EXPORT fszTargets |
| 136 | + FILE fszTargets.cmake |
| 137 | + NAMESPACE fsz:: |
| 138 | + DESTINATION ${CMAKE_INSTALL_LIBDIR}/cmake/fsz) |
| 139 | + |
| 140 | +configure_package_config_file(cmake/fszConfig.cmake.in |
| 141 | + ${CMAKE_CURRENT_BINARY_DIR}/fszConfig.cmake |
| 142 | + INSTALL_DESTINATION ${CMAKE_INSTALL_LIBDIR}/cmake/fsz) |
| 143 | +write_basic_package_version_file( |
| 144 | + ${CMAKE_CURRENT_BINARY_DIR}/fszConfigVersion.cmake |
| 145 | + VERSION ${PROJECT_VERSION} |
| 146 | + COMPATIBILITY SameMajorVersion) |
| 147 | +install(FILES |
| 148 | + ${CMAKE_CURRENT_BINARY_DIR}/fszConfig.cmake |
| 149 | + ${CMAKE_CURRENT_BINARY_DIR}/fszConfigVersion.cmake |
| 150 | + DESTINATION ${CMAKE_INSTALL_LIBDIR}/cmake/fsz) |
0 commit comments