1 # Distributed under the OSI-approved BSD 3-Clause License. See accompanying
2 # file Copyright.txt or https://cmake.org/licensing for details.
4 include(${CMAKE_ROOT}/Modules/CMakeDetermineCompiler.cmake)
5 include(${CMAKE_ROOT}/Modules/CMakeParseImplicitLinkInfo.cmake)
7 if( NOT ( ("${CMAKE_GENERATOR}" MATCHES "Make") OR
8 ("${CMAKE_GENERATOR}" MATCHES "Ninja") ) )
9 message(FATAL_ERROR "HIP language not currently supported by \"${CMAKE_GENERATOR}\" generator")
13 if(NOT CMAKE_HIP_COMPILER)
14 set(CMAKE_HIP_COMPILER_INIT NOTFOUND)
16 # prefer the environment variable HIPCXX
17 if(NOT $ENV{HIPCXX} STREQUAL "")
18 if("$ENV{HIPCXX}" MATCHES "hipcc")
20 "The HIPCXX environment variable is set to the hipcc wrapper:\n"
22 "This is not supported. Use Clang directly, or let CMake pick a default."
25 get_filename_component(CMAKE_HIP_COMPILER_INIT $ENV{HIPCXX} PROGRAM PROGRAM_ARGS CMAKE_HIP_FLAGS_ENV_INIT)
26 if(CMAKE_HIP_FLAGS_ENV_INIT)
27 set(CMAKE_HIP_COMPILER_ARG1 "${CMAKE_HIP_FLAGS_ENV_INIT}" CACHE STRING "Arguments to CXX compiler")
29 if(NOT EXISTS ${CMAKE_HIP_COMPILER_INIT})
30 message(FATAL_ERROR "Could not find compiler set in environment variable HIPCXX:\n$ENV{HIPCXX}.\n${CMAKE_HIP_COMPILER_INIT}")
34 # finally list compilers to try
35 if(NOT CMAKE_HIP_COMPILER_INIT)
36 set(CMAKE_HIP_COMPILER_LIST clang++)
38 # Look for the Clang coming with ROCm to support HIP.
39 execute_process(COMMAND hipconfig --hipclangpath
40 OUTPUT_VARIABLE _CMAKE_HIPCONFIG_CLANGPATH
41 RESULT_VARIABLE _CMAKE_HIPCONFIG_RESULT
43 if(_CMAKE_HIPCONFIG_RESULT EQUAL 0 AND EXISTS "${_CMAKE_HIPCONFIG_CLANGPATH}")
44 set(CMAKE_HIP_COMPILER_HINTS "${_CMAKE_HIPCONFIG_CLANGPATH}")
48 _cmake_find_compiler(HIP)
49 elseif(CMAKE_HIP_COMPILER MATCHES "hipcc")
51 "CMAKE_HIP_COMPILER is set to the hipcc wrapper:\n"
52 " ${CMAKE_HIP_COMPILER}\n"
53 "This is not supported. Use Clang directly, or let CMake pick a default."
56 _cmake_find_compiler_path(HIP)
59 mark_as_advanced(CMAKE_HIP_COMPILER)
61 # Build a small source file to identify the compiler.
62 if(NOT CMAKE_HIP_COMPILER_ID_RUN)
63 set(CMAKE_HIP_COMPILER_ID_RUN 1)
65 # Try to identify the compiler.
66 set(CMAKE_HIP_COMPILER_ID)
67 set(CMAKE_HIP_PLATFORM_ID)
68 file(READ ${CMAKE_ROOT}/Modules/CMakePlatformId.h.in
69 CMAKE_HIP_COMPILER_ID_PLATFORM_CONTENT)
71 list(APPEND CMAKE_HIP_COMPILER_ID_TEST_FLAGS_FIRST "-v")
73 include(${CMAKE_ROOT}/Modules/CMakeDetermineCompilerId.cmake)
74 CMAKE_DETERMINE_COMPILER_ID(HIP HIPFLAGS CMakeHIPCompilerId.hip)
76 _cmake_find_compiler_sysroot(HIP)
79 if(NOT CMAKE_HIP_COMPILER_ROCM_ROOT AND CMAKE_HIP_COMPILER_ID STREQUAL "Clang")
80 execute_process(COMMAND "${CMAKE_HIP_COMPILER}" -v -print-targets
81 OUTPUT_STRIP_TRAILING_WHITESPACE
82 RESULT_VARIABLE _CMAKE_HIP_COMPILER_RESULT
83 OUTPUT_VARIABLE _CMAKE_HIP_COMPILER_STDOUT
84 ERROR_VARIABLE _CMAKE_HIP_COMPILER_STDERR
87 if(_CMAKE_HIP_COMPILER_RESULT EQUAL 0 AND _CMAKE_HIP_COMPILER_STDERR MATCHES "Found HIP installation: *([^,]*)[,\n]")
88 set(CMAKE_HIP_COMPILER_ROCM_ROOT "${CMAKE_MATCH_1}")
89 file(TO_CMAKE_PATH "${CMAKE_HIP_COMPILER_ROCM_ROOT}" CMAKE_HIP_COMPILER_ROCM_ROOT)
92 if(NOT CMAKE_HIP_COMPILER_ROCM_ROOT)
94 COMMAND hipconfig --rocmpath
95 OUTPUT_VARIABLE _CMAKE_HIPCONFIG_ROCMPATH
96 RESULT_VARIABLE _CMAKE_HIPCONFIG_RESULT
98 if(_CMAKE_HIPCONFIG_RESULT EQUAL 0 AND EXISTS "${_CMAKE_HIPCONFIG_ROCMPATH}")
99 set(CMAKE_HIP_COMPILER_ROCM_ROOT "${_CMAKE_HIPCONFIG_ROCMPATH}")
102 if(NOT CMAKE_HIP_COMPILER_ROCM_ROOT)
103 message(FATAL_ERROR "Failed to find ROCm root directory.")
105 if(NOT EXISTS "${CMAKE_HIP_COMPILER_ROCM_ROOT}/lib/cmake/hip-lang/hip-lang-config.cmake")
107 "The ROCm root directory:\n"
108 " ${CMAKE_HIP_COMPILER_ROCM_ROOT}\n"
109 "does not contain the HIP runtime CMake package, expected at:\n"
110 " ${CMAKE_HIP_COMPILER_ROCM_ROOT}/lib/cmake/hip-lang/hip-lang-config.cmake\n"
114 if (NOT _CMAKE_TOOLCHAIN_LOCATION)
115 get_filename_component(_CMAKE_TOOLCHAIN_LOCATION "${CMAKE_HIP_COMPILER}" PATH)
118 set(_CMAKE_PROCESSING_LANGUAGE "HIP")
119 include(CMakeFindBinUtils)
120 include(Compiler/${CMAKE_HIP_COMPILER_ID}-FindBinUtils OPTIONAL)
121 unset(_CMAKE_PROCESSING_LANGUAGE)
123 if(CMAKE_HIP_COMPILER_SYSROOT)
124 string(CONCAT _SET_CMAKE_HIP_COMPILER_SYSROOT
125 "set(CMAKE_HIP_COMPILER_SYSROOT \"${CMAKE_HIP_COMPILER_SYSROOT}\")\n"
126 "set(CMAKE_COMPILER_SYSROOT \"${CMAKE_HIP_COMPILER_SYSROOT}\")")
128 set(_SET_CMAKE_HIP_COMPILER_SYSROOT "")
131 if(CMAKE_HIP_COMPILER_ARCHITECTURE_ID)
132 set(_SET_CMAKE_HIP_COMPILER_ARCHITECTURE_ID
133 "set(CMAKE_HIP_COMPILER_ARCHITECTURE_ID ${CMAKE_HIP_COMPILER_ARCHITECTURE_ID})")
135 set(_SET_CMAKE_HIP_COMPILER_ARCHITECTURE_ID "")
138 if(MSVC_HIP_ARCHITECTURE_ID)
139 set(SET_MSVC_HIP_ARCHITECTURE_ID
140 "set(MSVC_HIP_ARCHITECTURE_ID ${MSVC_HIP_ARCHITECTURE_ID})")
143 if(NOT DEFINED CMAKE_HIP_ARCHITECTURES)
144 # Use 'rocm_agent_enumerator' to get the current GPU architecture.
145 set(_CMAKE_HIP_ARCHITECTURES)
146 find_program(_CMAKE_HIP_ROCM_AGENT_ENUMERATOR
147 NAMES rocm_agent_enumerator
148 HINTS "${CMAKE_HIP_COMPILER_ROCM_ROOT}/bin"
150 if(_CMAKE_HIP_ROCM_AGENT_ENUMERATOR)
151 execute_process(COMMAND "${_CMAKE_HIP_ROCM_AGENT_ENUMERATOR}" -t GPU
152 RESULT_VARIABLE _CMAKE_ROCM_AGENT_ENUMERATOR_RESULT
153 OUTPUT_VARIABLE _CMAKE_ROCM_AGENT_ENUMERATOR_STDOUT
154 ERROR_VARIABLE _CMAKE_ROCM_AGENT_ENUMERATOR_STDERR
156 if(_CMAKE_ROCM_AGENT_ENUMERATOR_RESULT EQUAL 0)
157 separate_arguments(_hip_archs NATIVE_COMMAND "${_CMAKE_ROCM_AGENT_ENUMERATOR_STDOUT}")
158 foreach(_hip_arch ${_hip_archs})
159 if(_hip_arch STREQUAL "gfx000")
162 string(FIND ${_hip_arch} ":" pos)
163 if(NOT pos STREQUAL "-1")
164 string(SUBSTRING ${_hip_arch} 0 ${pos} _hip_arch)
166 list(APPEND _CMAKE_HIP_ARCHITECTURES "${_hip_arch}")
169 unset(_CMAKE_ROCM_AGENT_ENUMERATOR_RESULT)
170 unset(_CMAKE_ROCM_AGENT_ENUMERATOR_STDOUT)
171 unset(_CMAKE_ROCM_AGENT_ENUMERATOR_STDERR)
173 unset(_CMAKE_HIP_ROCM_AGENT_ENUMERATOR)
174 if(_CMAKE_HIP_ARCHITECTURES)
175 set(CMAKE_HIP_ARCHITECTURES "${_CMAKE_HIP_ARCHITECTURES}" CACHE STRING "HIP architectures")
176 elseif(CMAKE_HIP_COMPILER_PRODUCED_OUTPUT MATCHES " -target-cpu ([a-z0-9]+) ")
177 set(CMAKE_HIP_ARCHITECTURES "${CMAKE_MATCH_1}" CACHE STRING "HIP architectures")
179 message(FATAL_ERROR "Failed to find a default HIP architecture.")
181 unset(_CMAKE_HIP_ARCHITECTURES)
184 # configure variables set in this file for fast reload later on
185 configure_file(${CMAKE_ROOT}/Modules/CMakeHIPCompiler.cmake.in
186 ${CMAKE_PLATFORM_INFO_DIR}/CMakeHIPCompiler.cmake
189 set(CMAKE_HIP_COMPILER_ENV_VAR "HIPCXX")