CMakeLists.txt 1.9 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  1. cmake_minimum_required(VERSION 3.18)
  2. project(ProperDeviceLibraries CXX CUDA)
  3. set(CMAKE_CUDA_STANDARD 11)
  4. set(THREADS_PREFER_PTHREAD_FLAG ON)
  5. find_package(Threads)
  6. if(CMAKE_CUDA_COMPILER_VERSION VERSION_LESS 10.0.0)
  7. # cublas_device requires at least sm_35.
  8. set(CMAKE_CUDA_ARCHITECTURES 35)
  9. endif()
  10. add_executable(ProperDeviceLibraries main.cu)
  11. set_target_properties(ProperDeviceLibraries
  12. PROPERTIES CUDA_SEPARABLE_COMPILATION ON)
  13. add_library(UseThreadsMixed SHARED use_pthreads.cxx use_pthreads.cu)
  14. target_link_libraries(UseThreadsMixed Threads::Threads)
  15. add_library(UseThreadsCuda SHARED use_pthreads.cu)
  16. target_link_libraries(UseThreadsCuda Threads::Threads)
  17. target_link_libraries(ProperDeviceLibraries PRIVATE UseThreadsMixed UseThreadsCuda)
  18. if(THREADS_HAVE_PTHREAD_ARG AND CMAKE_USE_PTHREADS_INIT)
  19. add_library(UseExplicitPThreadsFlag SHARED use_pthreads.cu)
  20. target_compile_options(UseExplicitPThreadsFlag PUBLIC "-Xcompiler=-pthread")
  21. target_link_libraries(UseExplicitPThreadsFlag PUBLIC "-pthread")
  22. add_library(UseExplicitLThreadsFlag SHARED use_pthreads.cu)
  23. target_compile_options(UseExplicitLThreadsFlag PUBLIC "-Xcompiler=-pthread")
  24. target_link_libraries(UseExplicitLThreadsFlag PUBLIC "-lpthread")
  25. add_library(UseExplicitLongThreadsFlag SHARED use_pthreads.cu)
  26. target_link_libraries(UseExplicitLongThreadsFlag PUBLIC "--library pthread")
  27. target_link_libraries(ProperDeviceLibraries PRIVATE UseExplicitPThreadsFlag UseExplicitLThreadsFlag UseExplicitLongThreadsFlag)
  28. endif()
  29. if(CMAKE_CUDA_COMPILER_VERSION VERSION_LESS 10.0.0)
  30. #CUDA 10 removed the cublas_device library
  31. target_link_libraries(ProperDeviceLibraries PRIVATE cublas_device)
  32. endif()
  33. if(APPLE)
  34. # Help the static cuda runtime find the driver (libcuda.dyllib) at runtime.
  35. set_property(TARGET ProperDeviceLibraries PROPERTY BUILD_RPATH ${CMAKE_CUDA_IMPLICIT_LINK_DIRECTORIES})
  36. endif()