FindOpenACC.cmake 8.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257
  1. # Distributed under the OSI-approved BSD 3-Clause License. See accompanying
  2. # file Copyright.txt or https://cmake.org/licensing for details.
  3. #[=======================================================================[.rst:
  4. FindOpenACC
  5. -----------
  6. Detect OpenACC support by the compiler.
  7. This module can be used to detect OpenACC support in a compiler.
  8. If the compiler supports OpenACC, the flags required to compile with
  9. OpenACC support are returned in variables for the different languages.
  10. Currently, only PGI, GNU and Cray compilers are supported.
  11. Variables
  12. ^^^^^^^^^
  13. This module will set the following variables per language in your
  14. project, where ``<lang>`` is one of C, CXX, or Fortran:
  15. ``OpenACC_<lang>_FOUND``
  16. Variable indicating if OpenACC support for ``<lang>`` was detected.
  17. ``OpenACC_<lang>_FLAGS``
  18. OpenACC compiler flags for ``<lang>``, separated by spaces.
  19. The module will also try to provide the OpenACC version variables:
  20. ``OpenACC_<lang>_SPEC_DATE``
  21. Date of the OpenACC specification implemented by the ``<lang>`` compiler.
  22. ``OpenACC_<lang>_VERSION_MAJOR``
  23. Major version of OpenACC implemented by the ``<lang>`` compiler.
  24. ``OpenACC_<lang>_VERSION_MINOR``
  25. Minor version of OpenACC implemented by the ``<lang>`` compiler.
  26. ``OpenACC_<lang>_VERSION``
  27. OpenACC version implemented by the ``<lang>`` compiler.
  28. The specification date is formatted as given in the OpenACC standard:
  29. ``yyyymm`` where ``yyyy`` and ``mm`` represents the year and month of
  30. the OpenACC specification implemented by the ``<lang>`` compiler.
  31. Input Variables
  32. ^^^^^^^^^^^^^^^
  33. ``OpenACC_ACCEL_TARGET=<target>``
  34. If set, will the correct target accelerator flag set to the <target> will
  35. be returned with OpenACC_<lang>_FLAGS.
  36. #]=======================================================================]
  37. set(OpenACC_C_CXX_TEST_SOURCE
  38. "
  39. int main(){
  40. #ifdef _OPENACC
  41. return 0;
  42. #else
  43. breaks_on_purpose
  44. #endif
  45. }
  46. "
  47. )
  48. set(OpenACC_Fortran_TEST_SOURCE
  49. "
  50. program test
  51. #ifdef _OPENACC
  52. return 0;
  53. #else
  54. breaks_on_purpose
  55. #endif
  56. endprogram test
  57. "
  58. )
  59. set(OpenACC_C_CXX_CHECK_VERSION_SOURCE
  60. "
  61. #include <stdio.h>
  62. const char accver_str[] = { 'I', 'N', 'F', 'O', ':', 'O', 'p', 'e', 'n', 'A',
  63. 'C', 'C', '-', 'd', 'a', 't', 'e', '[',
  64. ('0' + ((_OPENACC/100000)%10)),
  65. ('0' + ((_OPENACC/10000)%10)),
  66. ('0' + ((_OPENACC/1000)%10)),
  67. ('0' + ((_OPENACC/100)%10)),
  68. ('0' + ((_OPENACC/10)%10)),
  69. ('0' + ((_OPENACC/1)%10)),
  70. ']', '\\0' };
  71. int main()
  72. {
  73. puts(accver_str);
  74. return 0;
  75. }
  76. ")
  77. set(OpenACC_Fortran_CHECK_VERSION_SOURCE
  78. "
  79. program acc_ver
  80. implicit none
  81. integer, parameter :: zero = ichar('0')
  82. character, dimension(25), parameter :: accver_str =&
  83. (/ 'I', 'N', 'F', 'O', ':', 'O', 'p', 'e', 'n', 'A', 'C', 'C', '-',&
  84. 'd', 'a', 't', 'e', '[',&
  85. char(zero + mod(_OPENACC/100000, 10)),&
  86. char(zero + mod(_OPENACC/10000, 10)),&
  87. char(zero + mod(_OPENACC/1000, 10)),&
  88. char(zero + mod(_OPENACC/100, 10)),&
  89. char(zero + mod(_OPENACC/10, 10)),&
  90. char(zero + mod(_OPENACC/1, 10)), ']' /)
  91. print *, accver_str
  92. end program acc_ver
  93. "
  94. )
  95. function(_OPENACC_WRITE_SOURCE_FILE LANG SRC_FILE_CONTENT_VAR SRC_FILE_NAME SRC_FILE_FULLPATH)
  96. set(WORK_DIR ${CMAKE_BINARY_DIR}${CMAKE_FILES_DIRECTORY}/FindOpenACC)
  97. if("${LANG}" STREQUAL "C")
  98. set(SRC_FILE "${WORK_DIR}/${SRC_FILE_NAME}.c")
  99. file(WRITE "${SRC_FILE}" "${OpenACC_C_CXX_${SRC_FILE_CONTENT_VAR}}")
  100. elseif("${LANG}" STREQUAL "CXX")
  101. set(SRC_FILE "${WORK_DIR}/${SRC_FILE_NAME}.cpp")
  102. file(WRITE "${SRC_FILE}" "${OpenACC_C_CXX_${SRC_FILE_CONTENT_VAR}}")
  103. elseif("${LANG}" STREQUAL "Fortran")
  104. set(SRC_FILE "${WORK_DIR}/${SRC_FILE_NAME}.F90")
  105. file(WRITE "${SRC_FILE}_in" "${OpenACC_Fortran_${SRC_FILE_CONTENT_VAR}}")
  106. configure_file("${SRC_FILE}_in" "${SRC_FILE}" @ONLY)
  107. endif()
  108. set(${SRC_FILE_FULLPATH} "${SRC_FILE}" PARENT_SCOPE)
  109. endfunction()
  110. function(_OPENACC_GET_FLAGS_CANDIDATE LANG FLAG_VAR)
  111. set(ACC_FLAG_PGI "-acc")
  112. set(ACC_FLAG_GNU "-fopenacc")
  113. set(ACC_FLAG_Cray "-h acc")
  114. if(DEFINED ACC_FLAG_${CMAKE_${LANG}_COMPILER_ID})
  115. set("${FLAG_VAR}" "${ACC_FLAG_${CMAKE_${LANG}_COMPILER_ID}}" PARENT_SCOPE)
  116. else()
  117. # Fall back to a few common flags.
  118. set("${FLAG_VAR}" ${ACC_FLAG_GNU} ${ACC_FLAG_PGI})
  119. endif()
  120. endfunction()
  121. function(_OPENACC_GET_ACCEL_TARGET_FLAG LANG TARGET FLAG_VAR)
  122. # Find target accelerator flags.
  123. set(ACC_TARGET_FLAG_PGI "-ta")
  124. if(DEFINED ACC_TARGET_FLAG_${CMAKE_${LANG}_COMPILER_ID})
  125. set("${FLAG_VAR}" "${ACC_TARGET_FLAG_${CMAKE_${LANG}_COMPILER_ID}}=${TARGET}" PARENT_SCOPE)
  126. endif()
  127. endfunction()
  128. function(_OPENACC_GET_VERBOSE_FLAG LANG FLAG_VAR)
  129. # Find compiler's verbose flag for OpenACC.
  130. set(ACC_VERBOSE_FLAG_PGI "-Minfo=accel")
  131. if(DEFINED ACC_VERBOSE_FLAG_${CMAKE_${LANG}_COMPILER_ID})
  132. set("${FLAG_VAR}" "${ACC_VERBOSE_FLAG_${CMAKE_${LANG}_COMPILER_ID}}" PARENT_SCOPE)
  133. endif()
  134. endfunction()
  135. function(_OPENACC_GET_FLAGS LANG FLAG_VAR)
  136. set(FLAG_CANDIDATES "")
  137. _OPENACC_GET_FLAGS_CANDIDATE("${LANG}" FLAG_CANDIDATES)
  138. _OPENACC_WRITE_SOURCE_FILE("${LANG}" "TEST_SOURCE" OpenACCTryFlag _OPENACC_TEST_SRC)
  139. foreach(FLAG IN LISTS FLAG_CANDIDATES)
  140. try_compile(OpenACC_FLAG_TEST_RESULT ${CMAKE_BINARY_DIR} ${_OPENACC_TEST_SRC}
  141. CMAKE_FLAGS "-DCOMPILE_DEFINITIONS:STRING=${FLAG}"
  142. OUTPUT_VARIABLE OpenACC_TRY_COMPILE_OUTPUT
  143. )
  144. if(OpenACC_FLAG_TEST_RESULT)
  145. set("${FLAG_VAR}" "${FLAG}")
  146. if(DEFINED OpenACC_ACCEL_TARGET)
  147. _OPENACC_GET_ACCEL_TARGET_FLAG("${LANG}" "${OpenACC_ACCEL_TARGET}" TARGET_FLAG)
  148. string(APPEND "${FLAG_VAR}" " ${TARGET_FLAG}")
  149. endif()
  150. if(CMAKE_VERBOSE_MAKEFILE)
  151. # -Minfo=accel prints out OpenACC's messages on optimizations.
  152. _OPENACC_GET_VERBOSE_FLAG("${LANG}" OpenACC_VERBOSE_FLAG)
  153. string(APPEND "${FLAG_VAR}" " ${OpenACC_VERBOSE_FLAG}")
  154. endif()
  155. set("${FLAG_VAR}" "${${FLAG_VAR}}" PARENT_SCOPE)
  156. break()
  157. endif()
  158. endforeach()
  159. endfunction()
  160. function(_OPENACC_GET_SPEC_DATE LANG SPEC_DATE)
  161. _OPENACC_WRITE_SOURCE_FILE("${LANG}" "CHECK_VERSION_SOURCE" OpenACCCheckVersion _OPENACC_TEST_SRC)
  162. set(BIN_FILE "${CMAKE_BINARY_DIR}${CMAKE_FILES_DIRECTORY}/FindOpenACC/accver_${LANG}.bin")
  163. try_compile(OpenACC_SPECTEST_${LANG} "${CMAKE_BINARY_DIR}" "${_OPENACC_TEST_SRC}"
  164. CMAKE_FLAGS "-DCOMPILE_DEFINITIONS:STRING=${OpenACC_${LANG}_FLAGS}"
  165. COPY_FILE ${BIN_FILE}
  166. OUTPUT_VARIABLE OUTPUT)
  167. if(${OpenACC_SPECTEST_${LANG}})
  168. file(STRINGS ${BIN_FILE} specstr LIMIT_COUNT 1 REGEX "INFO:OpenACC-date")
  169. set(regex_spec_date ".*INFO:OpenACC-date\\[0*([^]]*)\\].*")
  170. if("${specstr}" MATCHES "${regex_spec_date}")
  171. set(${SPEC_DATE} "${CMAKE_MATCH_1}" PARENT_SCOPE)
  172. endif()
  173. endif()
  174. endfunction()
  175. macro(_OPENACC_SET_VERSION_BY_SPEC_DATE LANG)
  176. set(OpenACC_SPEC_DATE_MAP
  177. # Combined versions, 2.5 onwards
  178. "201510=2.5"
  179. # 2013 08 is the corrected version.
  180. "201308=2.0"
  181. "201306=2.0"
  182. "201111=1.0"
  183. )
  184. string(REGEX MATCHALL "${OpenACC_${LANG}_SPEC_DATE}=([0-9]+)\\.([0-9]+)" _version_match "${OpenACC_SPEC_DATE_MAP}")
  185. if(NOT _version_match STREQUAL "")
  186. set(OpenACC_${LANG}_VERSION_MAJOR ${CMAKE_MATCH_1})
  187. set(OpenACC_${LANG}_VERSION_MINOR ${CMAKE_MATCH_2})
  188. set(OpenACC_${LANG}_VERSION "${OpenACC_${LANG}_VERSION_MAJOR}.${OpenACC_${LANG}_VERSION_MINOR}")
  189. else()
  190. unset(OpenACC_${LANG}_VERSION_MAJOR)
  191. unset(OpenACC_${LANG}_VERSION_MINOR)
  192. unset(OpenACC_${LANG}_VERSION)
  193. endif()
  194. unset(_version_match)
  195. unset(OpenACC_SPEC_DATE_MAP)
  196. endmacro()
  197. include(${CMAKE_CURRENT_LIST_DIR}/FindPackageHandleStandardArgs.cmake)
  198. foreach (LANG IN ITEMS C CXX Fortran)
  199. if(CMAKE_${LANG}_COMPILER_LOADED)
  200. set(OpenACC_${LANG}_FIND_QUIETLY ${OpenACC_FIND_QUIETLY})
  201. set(OpenACC_${LANG}_FIND_REQUIRED ${OpenACC_FIND_REQUIRED})
  202. set(OpenACC_${LANG}_FIND_VERSION ${OpenACC_FIND_VERSION})
  203. set(OpenACC_${LANG}_FIND_VERSION_EXACT ${OpenACC_FIND_VERSION_EXACT})
  204. if(NOT DEFINED OpenACC_${LANG}_FLAGS)
  205. _OPENACC_GET_FLAGS("${LANG}" OpenACC_${LANG}_FLAGS)
  206. endif()
  207. _OPENACC_GET_SPEC_DATE("${LANG}" OpenACC_${LANG}_SPEC_DATE)
  208. _OPENACC_SET_VERSION_BY_SPEC_DATE("${LANG}")
  209. find_package_handle_standard_args(OpenACC_${LANG}
  210. REQUIRED_VARS OpenACC_${LANG}_FLAGS
  211. VERSION_VAR OpenACC_${LANG}_VERSION
  212. )
  213. endif()
  214. endforeach()
  215. unset(OpenACC_C_CXX_TEST_SOURCE)
  216. unset(OpenACC_Fortran_TEST_SOURCE)
  217. unset(OpenACC_C_CXX_CHECK_VERSION_SOURCE)
  218. unset(OpenACC_Fortran_CHECK_VERSION_SOURCE)