Fix for building GPU backend on ROCm 5.0+

Change-Id: I32ad9be86d6a0467ccae555a1d0272813c905e97
This commit is contained in:
Nick Curtis
2022-04-07 17:48:03 -04:00
parent b4cc158981
commit ab0e5d5568

View File

@ -347,6 +347,10 @@ elseif(GPU_API STREQUAL "HIP")
target_link_libraries(gpu PRIVATE hip::host)
if(HIP_USE_DEVICE_SORT)
if(HIP_PLATFORM STREQUAL "amd")
# newer version of ROCm (5.1+) require c++14 for rocprim
set_property(TARGET gpu PROPERTY CXX_STANDARD 14)
endif()
# add hipCUB
target_include_directories(gpu PRIVATE ${HIP_ROOT_DIR}/../include)
target_compile_definitions(gpu PRIVATE -DUSE_HIP_DEVICE_SORT)