@@ -63,6 +63,7 @@ function(set_target_hip_arch target)
6363endfunction ()
6464
6565# Detect and enable CUDA
66+ STRING (REGEX REPLACE "\- std=[^ ]*" "" O2_GPU_CMAKE_CXX_FLAGS_NOSTD ${CMAKE_CXX_FLAGS} ) # Need to strip c++17 imposed by alidist defaults
6667if (ENABLE_CUDA)
6768 set (CMAKE_CUDA_STANDARD 17)
6869 set (CMAKE_CUDA_STANDARD_REQUIRED TRUE )
@@ -106,9 +107,7 @@ if(ENABLE_CUDA)
106107 endif ()
107108 endif ()
108109 if (CMAKE_CUDA_COMPILER)
109- # Forward CXX flags to CUDA C++ Host compiler (for warnings, gdb, etc.)
110- STRING (REGEX REPLACE "\- std=[^ ]*" "" CMAKE_CXX_FLAGS_NOSTD ${CMAKE_CXX_FLAGS} ) # Need to strip c++17 imposed by alidist defaults
111- set (CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -Xcompiler \" ${CMAKE_CXX_FLAGS_NOSTD} \" --expt-relaxed-constexpr --extended-lambda --allow-unsupported-compiler -Xptxas -v" )
110+ set (CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -Xcompiler \" ${O2_GPU_CMAKE_CXX_FLAGS_NOSTD} \" --expt-relaxed-constexpr --extended-lambda --allow-unsupported-compiler -Xptxas -v" )
112111 set (CMAKE_CUDA_FLAGS_DEBUG "${CMAKE_CUDA_FLAGS_DEBUG} -lineinfo -Xcompiler \" ${CMAKE_CXX_FLAGS_DEBUG} \" -Xptxas -O0 -Xcompiler -O0" )
113112 if (NOT CMAKE_BUILD_TYPE STREQUAL "DEBUG" )
114113 set (CMAKE_CUDA_FLAGS_${CMAKE_BUILD_TYPE} "${CMAKE_CUDA_FLAGS_${CMAKE_BUILD_TYPE} } -Xcompiler \" ${CMAKE_CXX_FLAGS_${CMAKE_BUILD_TYPE} }\" -Xptxas -O4 -Xcompiler -O4" )
@@ -262,7 +261,8 @@ if(ENABLE_HIP)
262261 string (REGEX REPLACE "(.*)bin/c\\ +\\ +\$ " "\\ 1" HIP_GCC_TOOLCHAIN_PATH "${CMAKE_CXX_COMPILER} " )
263262 set (O2_HIP_CMAKE_CXX_FLAGS "${O2_HIP_CMAKE_CXX_FLAGS} --gcc-toolchain=${HIP_GCC_TOOLCHAIN_PATH} " ) # -ffast-math disabled, since apparently it leads to miscompilation and crashes in FollowLooper kernel
264263 endif ()
265- set (CMAKE_HIP_FLAGS "${CMAKE_HIP_FLAGS} ${O2_HIP_CMAKE_CXX_FLAGS} " )
264+ set (CMAKE_HIP_FLAGS "${O2_GPU_CMAKE_CXX_FLAGS_NOSTD} ${CMAKE_HIP_FLAGS} ${O2_HIP_CMAKE_CXX_FLAGS} " )
265+ set (CMAKE_HIP_FLAGS_${CMAKE_BUILD_TYPE} "${CMAKE_CXX_FLAGS_${CMAKE_BUILD_TYPE} } ${CMAKE_HIP_FLAGS_${CMAKE_BUILD_TYPE} }" )
266266 else ()
267267 set (HIP_ENABLED OFF )
268268 endif ()
0 commit comments