| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849 |
- 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()
|