| cmake_minimum_required(VERSION 3.18) |
| project(ProperDeviceLibraries CXX CUDA) |
| |
| set(CMAKE_CUDA_STANDARD 11) |
| |
| set(THREADS_PREFER_PTHREAD_FLAG ON) |
| find_package(Threads) |
| |
| if(CMAKE_CUDA_COMPILER_VERSION VERSION_LESS 10.0.0) |
| # cublas_device requires at least sm_35. |
| set(CMAKE_CUDA_ARCHITECTURES 35) |
| endif() |
| |
| add_executable(ProperDeviceLibraries main.cu) |
| set_target_properties(ProperDeviceLibraries |
| PROPERTIES CUDA_SEPARABLE_COMPILATION ON) |
| |
| add_library(UseThreadsMixed SHARED use_pthreads.cxx use_pthreads.cu) |
| target_link_libraries(UseThreadsMixed Threads::Threads) |
| |
| add_library(UseThreadsCuda SHARED use_pthreads.cu) |
| target_link_libraries(UseThreadsCuda Threads::Threads) |
| |
| target_link_libraries(ProperDeviceLibraries PRIVATE UseThreadsMixed UseThreadsCuda) |
| |
| if(THREADS_HAVE_PTHREAD_ARG AND CMAKE_USE_PTHREADS_INIT) |
| add_library(UseExplicitPThreadsFlag SHARED use_pthreads.cu) |
| target_compile_options(UseExplicitPThreadsFlag PUBLIC "-Xcompiler=-pthread") |
| target_link_libraries(UseExplicitPThreadsFlag PUBLIC "-pthread") |
| |
| add_library(UseExplicitLThreadsFlag SHARED use_pthreads.cu) |
| target_compile_options(UseExplicitLThreadsFlag PUBLIC "-Xcompiler=-pthread") |
| target_link_libraries(UseExplicitLThreadsFlag PUBLIC "-lpthread") |
| |
| add_library(UseExplicitLongThreadsFlag SHARED use_pthreads.cu) |
| target_link_libraries(UseExplicitLongThreadsFlag PUBLIC "--library pthread") |
| |
| target_link_libraries(ProperDeviceLibraries PRIVATE UseExplicitPThreadsFlag UseExplicitLThreadsFlag UseExplicitLongThreadsFlag) |
| endif() |
| |
| if(CMAKE_CUDA_COMPILER_VERSION VERSION_LESS 10.0.0) |
| #CUDA 10 removed the cublas_device library |
| target_link_libraries(ProperDeviceLibraries PRIVATE cublas_device) |
| endif() |
| |
| if(APPLE) |
| # Help the static cuda runtime find the driver (libcuda.dyllib) at runtime. |
| set_property(TARGET ProperDeviceLibraries PROPERTY BUILD_RPATH ${CMAKE_CUDA_IMPLICIT_LINK_DIRECTORIES}) |
| endif() |
