CMakeCUDAInformation.cmake 5.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136
  1. # Distributed under the OSI-approved BSD 3-Clause License. See accompanying
  2. # file Copyright.txt or https://cmake.org/licensing for details.
  3. set(CMAKE_CUDA_OUTPUT_EXTENSION .o)
  4. set(CMAKE_INCLUDE_FLAG_CUDA "-I")
  5. # Load compiler-specific information.
  6. if(CMAKE_CUDA_COMPILER_ID)
  7. include(Compiler/${CMAKE_CUDA_COMPILER_ID}-CUDA OPTIONAL)
  8. endif()
  9. # load the system- and compiler specific files
  10. if(CMAKE_CUDA_COMPILER_ID)
  11. # load a hardware specific file, mostly useful for embedded compilers
  12. if(CMAKE_SYSTEM_PROCESSOR)
  13. include(Platform/${CMAKE_SYSTEM_NAME}-${CMAKE_CUDA_COMPILER_ID}-CUDA-${CMAKE_SYSTEM_PROCESSOR} OPTIONAL)
  14. endif()
  15. include(Platform/${CMAKE_SYSTEM_NAME}-${CMAKE_CUDA_COMPILER_ID}-CUDA OPTIONAL)
  16. endif()
  17. # for most systems a module is the same as a shared library
  18. # so unless the variable CMAKE_MODULE_EXISTS is set just
  19. # copy the values from the LIBRARY variables
  20. if(NOT CMAKE_MODULE_EXISTS)
  21. set(CMAKE_SHARED_MODULE_CUDA_FLAGS ${CMAKE_SHARED_LIBRARY_CUDA_FLAGS})
  22. set(CMAKE_SHARED_MODULE_CREATE_CUDA_FLAGS ${CMAKE_SHARED_LIBRARY_CREATE_CUDA_FLAGS})
  23. endif()
  24. # add the flags to the cache based
  25. # on the initial values computed in the platform/*.cmake files
  26. # use _INIT variables so that this only happens the first time
  27. # and you can set these flags in the cmake cache
  28. set(CMAKE_CUDA_FLAGS_INIT "$ENV{CUDAFLAGS} ${CMAKE_CUDA_FLAGS_INIT}")
  29. foreach(c "" _DEBUG _RELEASE _MINSIZEREL _RELWITHDEBINFO)
  30. string(STRIP "${CMAKE_CUDA_FLAGS${c}_INIT}" CMAKE_CUDA_FLAGS${c}_INIT)
  31. endforeach()
  32. set (CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS_INIT}" CACHE STRING
  33. "Flags used by the compiler during all build types.")
  34. if(NOT CMAKE_NOT_USING_CONFIG_FLAGS)
  35. set (CMAKE_CUDA_FLAGS_DEBUG "${CMAKE_CUDA_FLAGS_DEBUG_INIT}" CACHE STRING
  36. "Flags used by the compiler during debug builds.")
  37. set (CMAKE_CUDA_FLAGS_MINSIZEREL "${CMAKE_CUDA_FLAGS_MINSIZEREL_INIT}" CACHE STRING
  38. "Flags used by the compiler during release builds for minimum size.")
  39. set (CMAKE_CUDA_FLAGS_RELEASE "${CMAKE_CUDA_FLAGS_RELEASE_INIT}" CACHE STRING
  40. "Flags used by the compiler during release builds.")
  41. set (CMAKE_CUDA_FLAGS_RELWITHDEBINFO "${CMAKE_CUDA_FLAGS_RELWITHDEBINFO_INIT}" CACHE STRING
  42. "Flags used by the compiler during release builds with debug info.")
  43. endif()
  44. include(CMakeCommonLanguageInclude)
  45. # now define the following rules:
  46. # CMAKE_CUDA_CREATE_SHARED_LIBRARY
  47. # CMAKE_CUDA_CREATE_SHARED_MODULE
  48. # CMAKE_CUDA_COMPILE_OBJECT
  49. # CMAKE_CUDA_LINK_EXECUTABLE
  50. if(CMAKE_CUDA_HOST_COMPILER)
  51. set(CMAKE_CUDA_HOST_FLAGS "-ccbin=<CMAKE_CUDA_HOST_COMPILER>")
  52. else()
  53. set(CMAKE_CUDA_HOST_FLAGS "")
  54. endif()
  55. set(__IMPLICT_LINKS )
  56. foreach(dir ${CMAKE_CUDA_HOST_IMPLICIT_LINK_DIRECTORIES})
  57. string(APPEND __IMPLICT_LINKS " -L\"${dir}\"")
  58. endforeach()
  59. foreach(lib ${CMAKE_CUDA_HOST_IMPLICIT_LINK_LIBRARIES})
  60. if(${lib} MATCHES "/")
  61. string(APPEND __IMPLICT_LINKS " \"${lib}\"")
  62. else()
  63. string(APPEND __IMPLICT_LINKS " -l${lib}")
  64. endif()
  65. endforeach()
  66. # create a shared library
  67. if(NOT CMAKE_CUDA_CREATE_SHARED_LIBRARY)
  68. set(CMAKE_CUDA_CREATE_SHARED_LIBRARY
  69. "<CMAKE_CUDA_HOST_LINK_LAUNCHER> <CMAKE_SHARED_LIBRARY_CUDA_FLAGS> <LINK_FLAGS> <CMAKE_SHARED_LIBRARY_CREATE_CUDA_FLAGS> <SONAME_FLAG><TARGET_SONAME> -o <TARGET> <OBJECTS> <LINK_LIBRARIES>${__IMPLICT_LINKS}")
  70. endif()
  71. # create a shared module copy the shared library rule by default
  72. if(NOT CMAKE_CUDA_CREATE_SHARED_MODULE)
  73. set(CMAKE_CUDA_CREATE_SHARED_MODULE ${CMAKE_CUDA_CREATE_SHARED_LIBRARY})
  74. endif()
  75. # Create a static archive incrementally for large object file counts.
  76. if(NOT DEFINED CMAKE_CUDA_ARCHIVE_CREATE)
  77. set(CMAKE_CUDA_ARCHIVE_CREATE "<CMAKE_AR> qc <TARGET> <LINK_FLAGS> <OBJECTS>")
  78. endif()
  79. if(NOT DEFINED CMAKE_CUDA_ARCHIVE_APPEND)
  80. set(CMAKE_CUDA_ARCHIVE_APPEND "<CMAKE_AR> q <TARGET> <LINK_FLAGS> <OBJECTS>")
  81. endif()
  82. if(NOT DEFINED CMAKE_CUDA_ARCHIVE_FINISH)
  83. set(CMAKE_CUDA_ARCHIVE_FINISH "<CMAKE_RANLIB> <TARGET>")
  84. endif()
  85. # compile a cu file into an object file
  86. if(NOT CMAKE_CUDA_COMPILE_OBJECT)
  87. set(CMAKE_CUDA_COMPILE_OBJECT
  88. "<CMAKE_CUDA_COMPILER> ${CMAKE_CUDA_HOST_FLAGS} <DEFINES> <INCLUDES> <FLAGS> -x cu -c <SOURCE> -o <OBJECT>")
  89. #The Ninja generator uses the make file dependency files to determine what
  90. #files need to be recompiled. Unfortunately, nvcc doesn't support building
  91. #a source file and generating the dependencies of said file in a single
  92. #invocation. Instead we have to state that you need to chain two commands.
  93. #
  94. #The makefile generators uses the custom CMake dependency scanner, and thus
  95. #it is exempt from this logic.
  96. if(CMAKE_GENERATOR STREQUAL "Ninja")
  97. list(APPEND CMAKE_CUDA_COMPILE_OBJECT
  98. "<CMAKE_CUDA_COMPILER> ${CMAKE_CUDA_HOST_FLAGS} <DEFINES> <INCLUDES> <FLAGS> -x cu -M <SOURCE> -MT <OBJECT> -o $DEP_FILE")
  99. endif()
  100. endif()
  101. # compile a cu file into an executable
  102. if(NOT CMAKE_CUDA_LINK_EXECUTABLE)
  103. set(CMAKE_CUDA_LINK_EXECUTABLE
  104. "<CMAKE_CUDA_HOST_LINK_LAUNCHER> <CMAKE_CUDA_LINK_FLAGS> <LINK_FLAGS> <OBJECTS> -o <TARGET> <LINK_LIBRARIES>${__IMPLICT_LINKS}")
  105. endif()
  106. mark_as_advanced(
  107. CMAKE_CUDA_FLAGS
  108. CMAKE_CUDA_FLAGS_RELEASE
  109. CMAKE_CUDA_FLAGS_RELWITHDEBINFO
  110. CMAKE_CUDA_FLAGS_MINSIZEREL
  111. CMAKE_CUDA_FLAGS_DEBUG)
  112. set(CMAKE_CUDA_INFORMATION_LOADED 1)