CMakeCUDAInformation.cmake 8.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219
  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. # Set implicit links early so compiler-specific modules can use them.
  10. set(__IMPLICIT_LINKS)
  11. foreach(dir ${CMAKE_CUDA_HOST_IMPLICIT_LINK_DIRECTORIES})
  12. string(APPEND __IMPLICIT_LINKS " -L\"${dir}\"")
  13. endforeach()
  14. foreach(lib ${CMAKE_CUDA_HOST_IMPLICIT_LINK_LIBRARIES})
  15. if(${lib} MATCHES "/")
  16. string(APPEND __IMPLICIT_LINKS " \"${lib}\"")
  17. else()
  18. string(APPEND __IMPLICIT_LINKS " -l${lib}")
  19. endif()
  20. endforeach()
  21. # Load compiler-specific information.
  22. if(CMAKE_CUDA_COMPILER_ID)
  23. include(Compiler/${CMAKE_CUDA_COMPILER_ID}-CUDA OPTIONAL)
  24. endif()
  25. # load the system- and compiler specific files
  26. if(CMAKE_CUDA_COMPILER_ID)
  27. # load a hardware specific file, mostly useful for embedded compilers
  28. if(CMAKE_SYSTEM_PROCESSOR)
  29. include(Platform/${CMAKE_EFFECTIVE_SYSTEM_NAME}-${CMAKE_CUDA_COMPILER_ID}-CUDA-${CMAKE_SYSTEM_PROCESSOR} OPTIONAL)
  30. endif()
  31. include(Platform/${CMAKE_EFFECTIVE_SYSTEM_NAME}-${CMAKE_CUDA_COMPILER_ID}-CUDA OPTIONAL)
  32. endif()
  33. # This should be included before the _INIT variables are
  34. # used to initialize the cache. Since the rule variables
  35. # have if blocks on them, users can still define them here.
  36. # But, it should still be after the platform file so changes can
  37. # be made to those values.
  38. if(CMAKE_USER_MAKE_RULES_OVERRIDE)
  39. # Save the full path of the file so try_compile can use it.
  40. include(${CMAKE_USER_MAKE_RULES_OVERRIDE} RESULT_VARIABLE _override)
  41. set(CMAKE_USER_MAKE_RULES_OVERRIDE "${_override}")
  42. endif()
  43. if(CMAKE_USER_MAKE_RULES_OVERRIDE_CUDA)
  44. # Save the full path of the file so try_compile can use it.
  45. include(${CMAKE_USER_MAKE_RULES_OVERRIDE_CUDA} RESULT_VARIABLE _override)
  46. set(CMAKE_USER_MAKE_RULES_OVERRIDE_CUDA "${_override}")
  47. endif()
  48. if(NOT CMAKE_SHARED_LIBRARY_RUNTIME_CUDA_FLAG)
  49. set(CMAKE_SHARED_LIBRARY_RUNTIME_CUDA_FLAG ${CMAKE_SHARED_LIBRARY_RUNTIME_C_FLAG})
  50. endif()
  51. if(NOT CMAKE_SHARED_LIBRARY_RUNTIME_CUDA_FLAG_SEP)
  52. set(CMAKE_SHARED_LIBRARY_RUNTIME_CUDA_FLAG_SEP ${CMAKE_SHARED_LIBRARY_RUNTIME_C_FLAG_SEP})
  53. endif()
  54. if(NOT CMAKE_SHARED_LIBRARY_RPATH_LINK_CUDA_FLAG)
  55. set(CMAKE_SHARED_LIBRARY_RPATH_LINK_CUDA_FLAG ${CMAKE_SHARED_LIBRARY_RPATH_LINK_C_FLAG})
  56. endif()
  57. if(NOT DEFINED CMAKE_EXE_EXPORTS_CUDA_FLAG)
  58. set(CMAKE_EXE_EXPORTS_CUDA_FLAG ${CMAKE_EXE_EXPORTS_C_FLAG})
  59. endif()
  60. if(NOT DEFINED CMAKE_SHARED_LIBRARY_SONAME_CUDA_FLAG)
  61. set(CMAKE_SHARED_LIBRARY_SONAME_CUDA_FLAG ${CMAKE_SHARED_LIBRARY_SONAME_C_FLAG})
  62. endif()
  63. if(NOT CMAKE_EXECUTABLE_RUNTIME_CUDA_FLAG)
  64. set(CMAKE_EXECUTABLE_RUNTIME_CUDA_FLAG ${CMAKE_SHARED_LIBRARY_RUNTIME_CUDA_FLAG})
  65. endif()
  66. if(NOT CMAKE_EXECUTABLE_RUNTIME_CUDA_FLAG_SEP)
  67. set(CMAKE_EXECUTABLE_RUNTIME_CUDA_FLAG_SEP ${CMAKE_SHARED_LIBRARY_RUNTIME_CUDA_FLAG_SEP})
  68. endif()
  69. if(NOT CMAKE_EXECUTABLE_RPATH_LINK_CUDA_FLAG)
  70. set(CMAKE_EXECUTABLE_RPATH_LINK_CUDA_FLAG ${CMAKE_SHARED_LIBRARY_RPATH_LINK_CUDA_FLAG})
  71. endif()
  72. if(NOT DEFINED CMAKE_SHARED_LIBRARY_LINK_CUDA_WITH_RUNTIME_PATH)
  73. set(CMAKE_SHARED_LIBRARY_LINK_CUDA_WITH_RUNTIME_PATH ${CMAKE_SHARED_LIBRARY_LINK_C_WITH_RUNTIME_PATH})
  74. endif()
  75. # for most systems a module is the same as a shared library
  76. # so unless the variable CMAKE_MODULE_EXISTS is set just
  77. # copy the values from the LIBRARY variables
  78. if(NOT CMAKE_MODULE_EXISTS)
  79. set(CMAKE_SHARED_MODULE_CUDA_FLAGS ${CMAKE_SHARED_LIBRARY_CUDA_FLAGS})
  80. set(CMAKE_SHARED_MODULE_CREATE_CUDA_FLAGS ${CMAKE_SHARED_LIBRARY_CREATE_CUDA_FLAGS})
  81. endif()
  82. if(CMAKE_EXECUTABLE_FORMAT STREQUAL "ELF")
  83. if(NOT DEFINED CMAKE_CUDA_LINK_WHAT_YOU_USE_FLAG)
  84. set(CMAKE_CUDA_LINK_WHAT_YOU_USE_FLAG "LINKER:--no-as-needed")
  85. endif()
  86. if(NOT DEFINED CMAKE_LINK_WHAT_YOU_USE_CHECK)
  87. set(CMAKE_LINK_WHAT_YOU_USE_CHECK ldd -u -r)
  88. endif()
  89. endif()
  90. # add the flags to the cache based
  91. # on the initial values computed in the platform/*.cmake files
  92. # use _INIT variables so that this only happens the first time
  93. # and you can set these flags in the cmake cache
  94. set(CMAKE_CUDA_FLAGS_INIT "$ENV{CUDAFLAGS} ${CMAKE_CUDA_FLAGS_INIT}")
  95. cmake_initialize_per_config_variable(CMAKE_CUDA_FLAGS "Flags used by the CUDA compiler")
  96. if(CMAKE_CUDA_STANDARD_LIBRARIES_INIT)
  97. set(CMAKE_CUDA_STANDARD_LIBRARIES "${CMAKE_CUDA_STANDARD_LIBRARIES_INIT}"
  98. CACHE STRING "Libraries linked by default with all CUDA applications.")
  99. mark_as_advanced(CMAKE_CUDA_STANDARD_LIBRARIES)
  100. endif()
  101. if(NOT CMAKE_CUDA_COMPILER_LAUNCHER AND DEFINED ENV{CMAKE_CUDA_COMPILER_LAUNCHER})
  102. set(CMAKE_CUDA_COMPILER_LAUNCHER "$ENV{CMAKE_CUDA_COMPILER_LAUNCHER}"
  103. CACHE STRING "Compiler launcher for CUDA.")
  104. endif()
  105. include(CMakeCommonLanguageInclude)
  106. # now define the following rules:
  107. # CMAKE_CUDA_CREATE_SHARED_LIBRARY
  108. # CMAKE_CUDA_CREATE_SHARED_MODULE
  109. # CMAKE_CUDA_COMPILE_WHOLE_COMPILATION
  110. # CMAKE_CUDA_COMPILE_PTX_COMPILATION
  111. # CMAKE_CUDA_COMPILE_SEPARABLE_COMPILATION
  112. # CMAKE_CUDA_LINK_EXECUTABLE
  113. # create a shared library
  114. if(NOT CMAKE_CUDA_CREATE_SHARED_LIBRARY)
  115. set(CMAKE_CUDA_CREATE_SHARED_LIBRARY
  116. "<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>${__IMPLICIT_LINKS}")
  117. endif()
  118. # create a shared module copy the shared library rule by default
  119. if(NOT CMAKE_CUDA_CREATE_SHARED_MODULE)
  120. set(CMAKE_CUDA_CREATE_SHARED_MODULE ${CMAKE_CUDA_CREATE_SHARED_LIBRARY})
  121. endif()
  122. # Create a static archive incrementally for large object file counts.
  123. if(NOT DEFINED CMAKE_CUDA_ARCHIVE_CREATE)
  124. set(CMAKE_CUDA_ARCHIVE_CREATE "<CMAKE_AR> qc <TARGET> <LINK_FLAGS> <OBJECTS>")
  125. endif()
  126. if(NOT DEFINED CMAKE_CUDA_ARCHIVE_APPEND)
  127. set(CMAKE_CUDA_ARCHIVE_APPEND "<CMAKE_AR> q <TARGET> <LINK_FLAGS> <OBJECTS>")
  128. endif()
  129. if(NOT DEFINED CMAKE_CUDA_ARCHIVE_FINISH)
  130. set(CMAKE_CUDA_ARCHIVE_FINISH "<CMAKE_RANLIB> <TARGET>")
  131. endif()
  132. #Specify how to compile when ptx has been requested
  133. if(NOT CMAKE_CUDA_COMPILE_PTX_COMPILATION)
  134. set(CMAKE_CUDA_COMPILE_PTX_COMPILATION
  135. "<CMAKE_CUDA_COMPILER> ${_CMAKE_CUDA_EXTRA_FLAGS} <DEFINES> <INCLUDES> <FLAGS> ${_CMAKE_COMPILE_AS_CUDA_FLAG} ${_CMAKE_CUDA_PTX_FLAG} <SOURCE> -o <OBJECT>")
  136. endif()
  137. #Specify how to compile when separable compilation has been requested
  138. if(NOT CMAKE_CUDA_COMPILE_SEPARABLE_COMPILATION)
  139. set(CMAKE_CUDA_COMPILE_SEPARABLE_COMPILATION
  140. "<CMAKE_CUDA_COMPILER> ${_CMAKE_CUDA_EXTRA_FLAGS} <DEFINES> <INCLUDES> <FLAGS> ${_CMAKE_COMPILE_AS_CUDA_FLAG} ${_CMAKE_CUDA_DEVICE_CODE} <SOURCE> -o <OBJECT>")
  141. endif()
  142. #Specify how to compile when whole compilation has been requested
  143. if(NOT CMAKE_CUDA_COMPILE_WHOLE_COMPILATION)
  144. set(CMAKE_CUDA_COMPILE_WHOLE_COMPILATION
  145. "<CMAKE_CUDA_COMPILER> ${_CMAKE_CUDA_EXTRA_FLAGS} <DEFINES> <INCLUDES> <FLAGS> ${_CMAKE_COMPILE_AS_CUDA_FLAG} -c <SOURCE> -o <OBJECT>")
  146. endif()
  147. # compile a cu file into an executable
  148. if(NOT CMAKE_CUDA_LINK_EXECUTABLE)
  149. set(CMAKE_CUDA_LINK_EXECUTABLE
  150. "<CMAKE_CUDA_HOST_LINK_LAUNCHER> <LINK_FLAGS> <OBJECTS> -o <TARGET> <LINK_LIBRARIES>${__IMPLICIT_LINKS}")
  151. endif()
  152. # Add implicit host link directories that contain device libraries
  153. # to the device link line.
  154. set(__IMPLICIT_DLINK_DIRS ${CMAKE_CUDA_IMPLICIT_LINK_DIRECTORIES})
  155. if(__IMPLICIT_DLINK_DIRS)
  156. list(REMOVE_ITEM __IMPLICIT_DLINK_DIRS ${CMAKE_CUDA_HOST_IMPLICIT_LINK_DIRECTORIES})
  157. endif()
  158. set(__IMPLICIT_DLINK_FLAGS)
  159. foreach(dir ${__IMPLICIT_DLINK_DIRS})
  160. if(EXISTS "${dir}/libcurand_static.a")
  161. string(APPEND __IMPLICIT_DLINK_FLAGS " -L\"${dir}\"")
  162. endif()
  163. endforeach()
  164. unset(__IMPLICIT_DLINK_DIRS)
  165. #These are used when linking relocatable (dc) cuda code
  166. if(NOT CMAKE_CUDA_DEVICE_LINK_LIBRARY)
  167. set(CMAKE_CUDA_DEVICE_LINK_LIBRARY
  168. "<CMAKE_CUDA_COMPILER> ${_CMAKE_CUDA_EXTRA_FLAGS} <LANGUAGE_COMPILE_FLAGS> <LINK_FLAGS> ${CMAKE_CUDA_COMPILE_OPTIONS_PIC} ${_CMAKE_CUDA_EXTRA_DEVICE_LINK_FLAGS} -shared -dlink <OBJECTS> -o <TARGET> <LINK_LIBRARIES>${__IMPLICIT_DLINK_FLAGS}")
  169. endif()
  170. if(NOT CMAKE_CUDA_DEVICE_LINK_EXECUTABLE)
  171. set(CMAKE_CUDA_DEVICE_LINK_EXECUTABLE
  172. "<CMAKE_CUDA_COMPILER> ${_CMAKE_CUDA_EXTRA_FLAGS} <LANGUAGE_COMPILE_FLAGS> <LINK_FLAGS> ${CMAKE_CUDA_COMPILE_OPTIONS_PIC} ${_CMAKE_CUDA_EXTRA_DEVICE_LINK_FLAGS} -shared -dlink <OBJECTS> -o <TARGET> <LINK_LIBRARIES>${__IMPLICIT_DLINK_FLAGS}")
  173. endif()
  174. # Used when device linking is handled by CMake.
  175. if(NOT CMAKE_CUDA_DEVICE_LINK_COMPILE)
  176. set(CMAKE_CUDA_DEVICE_LINK_COMPILE "<CMAKE_CUDA_COMPILER> ${_CMAKE_CUDA_EXTRA_FLAGS} <FLAGS> -D__CUDA_INCLUDE_COMPILER_INTERNAL_HEADERS__ -D__NV_EXTRA_INITIALIZATION=\"\" -D__NV_EXTRA_FINALIZATION=\"\" -DREGISTERLINKBINARYFILE=\\\"<REGISTER_FILE>\\\" -DFATBINFILE=\\\"<FATBINARY>\\\" ${_CMAKE_COMPILE_AS_CUDA_FLAG} -c \"${CMAKE_CUDA_COMPILER_TOOLKIT_LIBRARY_ROOT}/bin/crt/link.stub\" -o <OBJECT>")
  177. endif()
  178. unset(__IMPLICIT_DLINK_FLAGS)
  179. set(CMAKE_CUDA_INFORMATION_LOADED 1)