CMakeCUDAInformation.cmake 8.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213
  1. # Distributed under the OSI-approved BSD 3-Clause License. See accompanying
  2. # file Copyright.txt or https://cmake.org/licensing for details.
  3. if(UNIX)
  4. set(CMAKE_CUDA_OUTPUT_EXTENSION .o)
  5. else()
  6. set(CMAKE_CUDA_OUTPUT_EXTENSION .obj)
  7. endif()
  8. set(CMAKE_INCLUDE_FLAG_CUDA "-I")
  9. # Load compiler-specific information.
  10. if(CMAKE_CUDA_COMPILER_ID)
  11. include(Compiler/${CMAKE_CUDA_COMPILER_ID}-CUDA OPTIONAL)
  12. endif()
  13. # load the system- and compiler specific files
  14. if(CMAKE_CUDA_COMPILER_ID)
  15. # load a hardware specific file, mostly useful for embedded compilers
  16. if(CMAKE_SYSTEM_PROCESSOR)
  17. include(Platform/${CMAKE_EFFECTIVE_SYSTEM_NAME}-${CMAKE_CUDA_COMPILER_ID}-CUDA-${CMAKE_SYSTEM_PROCESSOR} OPTIONAL)
  18. endif()
  19. include(Platform/${CMAKE_EFFECTIVE_SYSTEM_NAME}-${CMAKE_CUDA_COMPILER_ID}-CUDA OPTIONAL)
  20. endif()
  21. if(NOT CMAKE_SHARED_LIBRARY_RUNTIME_CUDA_FLAG)
  22. set(CMAKE_SHARED_LIBRARY_RUNTIME_CUDA_FLAG ${CMAKE_SHARED_LIBRARY_RUNTIME_C_FLAG})
  23. endif()
  24. if(NOT CMAKE_SHARED_LIBRARY_RUNTIME_CUDA_FLAG_SEP)
  25. set(CMAKE_SHARED_LIBRARY_RUNTIME_CUDA_FLAG_SEP ${CMAKE_SHARED_LIBRARY_RUNTIME_C_FLAG_SEP})
  26. endif()
  27. if(NOT CMAKE_SHARED_LIBRARY_RPATH_LINK_CUDA_FLAG)
  28. set(CMAKE_SHARED_LIBRARY_RPATH_LINK_CUDA_FLAG ${CMAKE_SHARED_LIBRARY_RPATH_LINK_C_FLAG})
  29. endif()
  30. if(NOT DEFINED CMAKE_EXE_EXPORTS_CUDA_FLAG)
  31. set(CMAKE_EXE_EXPORTS_CUDA_FLAG ${CMAKE_EXE_EXPORTS_C_FLAG})
  32. endif()
  33. if(NOT DEFINED CMAKE_SHARED_LIBRARY_SONAME_CUDA_FLAG)
  34. set(CMAKE_SHARED_LIBRARY_SONAME_CUDA_FLAG ${CMAKE_SHARED_LIBRARY_SONAME_C_FLAG})
  35. endif()
  36. if(NOT CMAKE_EXECUTABLE_RUNTIME_CUDA_FLAG)
  37. set(CMAKE_EXECUTABLE_RUNTIME_CUDA_FLAG ${CMAKE_SHARED_LIBRARY_RUNTIME_CUDA_FLAG})
  38. endif()
  39. if(NOT CMAKE_EXECUTABLE_RUNTIME_CUDA_FLAG_SEP)
  40. set(CMAKE_EXECUTABLE_RUNTIME_CUDA_FLAG_SEP ${CMAKE_SHARED_LIBRARY_RUNTIME_CUDA_FLAG_SEP})
  41. endif()
  42. if(NOT CMAKE_EXECUTABLE_RPATH_LINK_CUDA_FLAG)
  43. set(CMAKE_EXECUTABLE_RPATH_LINK_CUDA_FLAG ${CMAKE_SHARED_LIBRARY_RPATH_LINK_CUDA_FLAG})
  44. endif()
  45. if(NOT DEFINED CMAKE_SHARED_LIBRARY_LINK_CUDA_WITH_RUNTIME_PATH)
  46. set(CMAKE_SHARED_LIBRARY_LINK_CUDA_WITH_RUNTIME_PATH ${CMAKE_SHARED_LIBRARY_LINK_C_WITH_RUNTIME_PATH})
  47. endif()
  48. # for most systems a module is the same as a shared library
  49. # so unless the variable CMAKE_MODULE_EXISTS is set just
  50. # copy the values from the LIBRARY variables
  51. if(NOT CMAKE_MODULE_EXISTS)
  52. set(CMAKE_SHARED_MODULE_CUDA_FLAGS ${CMAKE_SHARED_LIBRARY_CUDA_FLAGS})
  53. set(CMAKE_SHARED_MODULE_CREATE_CUDA_FLAGS ${CMAKE_SHARED_LIBRARY_CREATE_CUDA_FLAGS})
  54. endif()
  55. # add the flags to the cache based
  56. # on the initial values computed in the platform/*.cmake files
  57. # use _INIT variables so that this only happens the first time
  58. # and you can set these flags in the cmake cache
  59. set(CMAKE_CUDA_FLAGS_INIT "$ENV{CUDAFLAGS} ${CMAKE_CUDA_FLAGS_INIT}")
  60. cmake_initialize_per_config_variable(CMAKE_CUDA_FLAGS "Flags used by the CUDA compiler")
  61. if(CMAKE_CUDA_STANDARD_LIBRARIES_INIT)
  62. set(CMAKE_CUDA_STANDARD_LIBRARIES "${CMAKE_CUDA_STANDARD_LIBRARIES_INIT}"
  63. CACHE STRING "Libraries linked by default with all CUDA applications.")
  64. mark_as_advanced(CMAKE_CUDA_STANDARD_LIBRARIES)
  65. endif()
  66. if(NOT CMAKE_CUDA_COMPILER_LAUNCHER AND DEFINED ENV{CMAKE_CUDA_COMPILER_LAUNCHER})
  67. set(CMAKE_CUDA_COMPILER_LAUNCHER "$ENV{CMAKE_CUDA_COMPILER_LAUNCHER}"
  68. CACHE STRING "Compiler launcher for CUDA.")
  69. endif()
  70. include(CMakeCommonLanguageInclude)
  71. # now define the following rules:
  72. # CMAKE_CUDA_CREATE_SHARED_LIBRARY
  73. # CMAKE_CUDA_CREATE_SHARED_MODULE
  74. # CMAKE_CUDA_COMPILE_WHOLE_COMPILATION
  75. # CMAKE_CUDA_COMPILE_PTX_COMPILATION
  76. # CMAKE_CUDA_COMPILE_SEPARABLE_COMPILATION
  77. # CMAKE_CUDA_LINK_EXECUTABLE
  78. if(CMAKE_CUDA_HOST_COMPILER)
  79. set(CMAKE_CUDA_HOST_FLAGS "-ccbin=<CMAKE_CUDA_HOST_COMPILER>")
  80. else()
  81. set(CMAKE_CUDA_HOST_FLAGS "")
  82. endif()
  83. set(__IMPLICT_LINKS )
  84. foreach(dir ${CMAKE_CUDA_HOST_IMPLICIT_LINK_DIRECTORIES})
  85. string(APPEND __IMPLICT_LINKS " -L\"${dir}\"")
  86. endforeach()
  87. foreach(lib ${CMAKE_CUDA_HOST_IMPLICIT_LINK_LIBRARIES})
  88. if(${lib} MATCHES "/")
  89. string(APPEND __IMPLICT_LINKS " \"${lib}\"")
  90. else()
  91. string(APPEND __IMPLICT_LINKS " -l${lib}")
  92. endif()
  93. endforeach()
  94. # create a shared library
  95. if(NOT CMAKE_CUDA_CREATE_SHARED_LIBRARY)
  96. set(CMAKE_CUDA_CREATE_SHARED_LIBRARY
  97. "<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}")
  98. endif()
  99. # create a shared module copy the shared library rule by default
  100. if(NOT CMAKE_CUDA_CREATE_SHARED_MODULE)
  101. set(CMAKE_CUDA_CREATE_SHARED_MODULE ${CMAKE_CUDA_CREATE_SHARED_LIBRARY})
  102. endif()
  103. # Create a static archive incrementally for large object file counts.
  104. if(NOT DEFINED CMAKE_CUDA_ARCHIVE_CREATE)
  105. set(CMAKE_CUDA_ARCHIVE_CREATE "<CMAKE_AR> qc <TARGET> <LINK_FLAGS> <OBJECTS>")
  106. endif()
  107. if(NOT DEFINED CMAKE_CUDA_ARCHIVE_APPEND)
  108. set(CMAKE_CUDA_ARCHIVE_APPEND "<CMAKE_AR> q <TARGET> <LINK_FLAGS> <OBJECTS>")
  109. endif()
  110. if(NOT DEFINED CMAKE_CUDA_ARCHIVE_FINISH)
  111. set(CMAKE_CUDA_ARCHIVE_FINISH "<CMAKE_RANLIB> <TARGET>")
  112. endif()
  113. #Specify how to compile when ptx has been requested
  114. if(NOT CMAKE_CUDA_COMPILE_PTX_COMPILATION)
  115. set(CMAKE_CUDA_COMPILE_PTX_COMPILATION
  116. "<CMAKE_CUDA_COMPILER> ${CMAKE_CUDA_HOST_FLAGS} <DEFINES> <INCLUDES> <FLAGS> -x cu -ptx <SOURCE> -o <OBJECT>")
  117. endif()
  118. #Specify how to compile when separable compilation has been requested
  119. if(NOT CMAKE_CUDA_COMPILE_SEPARABLE_COMPILATION)
  120. set(CMAKE_CUDA_COMPILE_SEPARABLE_COMPILATION
  121. "<CMAKE_CUDA_COMPILER> ${CMAKE_CUDA_HOST_FLAGS} <DEFINES> <INCLUDES> <FLAGS> -x cu -dc <SOURCE> -o <OBJECT>")
  122. endif()
  123. #Specify how to compile when whole compilation has been requested
  124. if(NOT CMAKE_CUDA_COMPILE_WHOLE_COMPILATION)
  125. set(CMAKE_CUDA_COMPILE_WHOLE_COMPILATION
  126. "<CMAKE_CUDA_COMPILER> ${CMAKE_CUDA_HOST_FLAGS} <DEFINES> <INCLUDES> <FLAGS> -x cu -c <SOURCE> -o <OBJECT>")
  127. endif()
  128. if(CMAKE_GENERATOR STREQUAL "Ninja")
  129. set(CMAKE_CUDA_COMPILE_DEPENDENCY_DETECTION
  130. "<CMAKE_CUDA_COMPILER> ${CMAKE_CUDA_HOST_FLAGS} <DEFINES> <INCLUDES> <FLAGS> -x cu -M <SOURCE> -MT <OBJECT> -o $DEP_FILE")
  131. #The Ninja generator uses the make file dependency files to determine what
  132. #files need to be recompiled. Unfortunately, nvcc doesn't support building
  133. #a source file and generating the dependencies of said file in a single
  134. #invocation. Instead we have to state that you need to chain two commands.
  135. #
  136. #The makefile generators uses the custom CMake dependency scanner, and thus
  137. #it is exempt from this logic.
  138. list(APPEND CMAKE_CUDA_COMPILE_PTX_COMPILATION "${CMAKE_CUDA_COMPILE_DEPENDENCY_DETECTION}")
  139. list(APPEND CMAKE_CUDA_COMPILE_SEPARABLE_COMPILATION "${CMAKE_CUDA_COMPILE_DEPENDENCY_DETECTION}")
  140. list(APPEND CMAKE_CUDA_COMPILE_WHOLE_COMPILATION "${CMAKE_CUDA_COMPILE_DEPENDENCY_DETECTION}")
  141. endif()
  142. # compile a cu file into an executable
  143. if(NOT CMAKE_CUDA_LINK_EXECUTABLE)
  144. set(CMAKE_CUDA_LINK_EXECUTABLE
  145. "<CMAKE_CUDA_HOST_LINK_LAUNCHER> <LINK_FLAGS> <OBJECTS> -o <TARGET> <LINK_LIBRARIES>${__IMPLICT_LINKS}")
  146. endif()
  147. if( CMAKE_CUDA_COMPILER_ID STREQUAL "NVIDIA" AND
  148. CMAKE_CUDA_COMPILER_VERSION VERSION_GREATER_EQUAL "8.0.0")
  149. set(_CMAKE_CUDA_EXTRA_DEVICE_LINK_FLAGS "-Wno-deprecated-gpu-targets")
  150. else()
  151. set(_CMAKE_CUDA_EXTRA_DEVICE_LINK_FLAGS "")
  152. endif()
  153. # Add implicit host link directories that contain device libraries
  154. # to the device link line.
  155. set(__IMPLICT_DLINK_DIRS ${CMAKE_CUDA_IMPLICIT_LINK_DIRECTORIES})
  156. if(__IMPLICT_DLINK_DIRS)
  157. list(REMOVE_ITEM __IMPLICT_DLINK_DIRS ${CMAKE_CUDA_HOST_IMPLICIT_LINK_DIRECTORIES})
  158. endif()
  159. set(__IMPLICT_DLINK_FLAGS )
  160. foreach(dir ${__IMPLICT_DLINK_DIRS})
  161. if(EXISTS "${dir}/libcurand_static.a")
  162. string(APPEND __IMPLICT_DLINK_FLAGS " -L\"${dir}\"")
  163. endif()
  164. endforeach()
  165. unset(__IMPLICT_DLINK_DIRS)
  166. #These are used when linking relocatable (dc) cuda code
  167. if(NOT CMAKE_CUDA_DEVICE_LINK_LIBRARY)
  168. set(CMAKE_CUDA_DEVICE_LINK_LIBRARY
  169. "<CMAKE_CUDA_COMPILER> ${CMAKE_CUDA_HOST_FLAGS} <LANGUAGE_COMPILE_FLAGS> ${CMAKE_CUDA_COMPILE_OPTIONS_PIC} ${_CMAKE_CUDA_EXTRA_DEVICE_LINK_FLAGS} -shared -dlink <OBJECTS> -o <TARGET> <LINK_LIBRARIES>${__IMPLICT_DLINK_FLAGS}")
  170. endif()
  171. if(NOT CMAKE_CUDA_DEVICE_LINK_EXECUTABLE)
  172. set(CMAKE_CUDA_DEVICE_LINK_EXECUTABLE
  173. "<CMAKE_CUDA_COMPILER> ${CMAKE_CUDA_HOST_FLAGS} <FLAGS> ${CMAKE_CUDA_COMPILE_OPTIONS_PIC} ${_CMAKE_CUDA_EXTRA_DEVICE_LINK_FLAGS} -shared -dlink <OBJECTS> -o <TARGET> <LINK_LIBRARIES>${__IMPLICT_DLINK_FLAGS}")
  174. endif()
  175. unset(_CMAKE_CUDA_EXTRA_DEVICE_LINK_FLAGS)
  176. unset(__IMPLICT_DLINK_FLAGS)
  177. set(CMAKE_CUDA_INFORMATION_LOADED 1)