Skip to content

Commit 623391b

Browse files
committed
GPU CMake: Force using alidist GCC for host compilation of CUDA/HIP code
1 parent 999ac71 commit 623391b

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

dependencies/FindO2GPU.cmake

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -125,6 +125,8 @@ if(ENABLE_CUDA)
125125
if(GPUCA_CUDA_GCCBIN)
126126
message(STATUS "Using as CUDA GCC version: ${GPUCA_CUDA_GCCBIN}")
127127
set(CMAKE_CUDA_HOST_COMPILER "${GPUCA_CUDA_GCCBIN}")
128+
elseif(DEFINED ENV{GCC_TOOLCHAIN_ROOT})
129+
set(CMAKE_CUDA_HOST_COMPILER "$ENV{GCC_TOOLCHAIN_ROOT}/bin/gcc")
128130
endif()
129131
enable_language(CUDA)
130132
get_property(LANGUAGES GLOBAL PROPERTY ENABLED_LANGUAGES)
@@ -278,6 +280,12 @@ if(ENABLE_HIP)
278280
endif()
279281
endif()
280282
if (CMAKE_HIP_COMPILER)
283+
if(GPUCA_HIP_GCCBIN)
284+
message(STATUS "Using as HIP GCC version: ${GPUCA_HIP_GCCBIN}")
285+
set(CMAKE_HIP_HOST_COMPILER "${GPUCA_HIP_GCCBIN}")
286+
elseif(DEFINED ENV{GCC_TOOLCHAIN_ROOT})
287+
set(CMAKE_HIP_HOST_COMPILER "$ENV{GCC_TOOLCHAIN_ROOT}/bin/gcc")
288+
endif()
281289
enable_language(HIP)
282290
message(STATUS "HIP language enabled: ${CMAKE_HIP_COMPILER}")
283291
endif()

0 commit comments

Comments
 (0)