mirror of
https://github.com/Kitware/CMake.git
synced 2026-05-07 22:59:56 -05:00
Brad King
81237523b7
Since commit 2def6a874b (CUDA: Add support for CUBIN, FATBIN, and
OPTIXIR compilation, 2023-01-27) added this test, it has failed when
executed with a CUDA toolkit that does not support the true native
architecture of the GPU. Detect this case and skip the test.
30 lines
1.2 KiB
CMake
30 lines
1.2 KiB
CMake
cmake_minimum_required(VERSION 3.18)
|
|
unset(ENV{CMAKE_CUDA_ARCHITECTURES_NATIVE_CLAMP}) # CUBIN needs true native arch
|
|
project(CudaCUBIN LANGUAGES CUDA)
|
|
|
|
set(CMAKE_CUDA_ARCHITECTURES all-major)
|
|
|
|
# CUBIN needs the true native arch to be supported by the CUDA toolkit.
|
|
set(unavailable_native_archs "${CMAKE_CUDA_ARCHITECTURES_NATIVE}")
|
|
list(REMOVE_ITEM unavailable_native_archs ${CMAKE_CUDA_ARCHITECTURES_ALL})
|
|
if(unavailable_native_archs)
|
|
add_executable(CudaOnlyCUBIN main_no_native_archs.cu)
|
|
return()
|
|
endif()
|
|
|
|
add_library(CudaCUBIN OBJECT kernelA.cu kernelB.cu kernelC.cu)
|
|
set_property(TARGET CudaCUBIN PROPERTY CUDA_CUBIN_COMPILATION ON)
|
|
set_property(TARGET CudaCUBIN PROPERTY CUDA_ARCHITECTURES native)
|
|
|
|
add_executable(CudaOnlyCUBIN main.cu)
|
|
target_compile_features(CudaOnlyCUBIN PRIVATE cuda_std_11)
|
|
target_compile_definitions(CudaOnlyCUBIN PRIVATE "CUBIN_FILE_PATHS=\"$<JOIN:$<TARGET_OBJECTS:CudaCUBIN>,~_~>\"")
|
|
|
|
find_package(CUDAToolkit REQUIRED)
|
|
target_link_libraries(CudaOnlyCUBIN PRIVATE CUDA::cuda_driver)
|
|
|
|
if(APPLE)
|
|
# Help the static cuda runtime find the driver (libcuda.dyllib) at runtime.
|
|
set_property(TARGET CudaOnlyCUBIN PROPERTY BUILD_RPATH ${CMAKE_CUDA_IMPLICIT_LINK_DIRECTORIES})
|
|
endif()
|