Add option -fhip-lambda-host-device for HIP-Clang (#518)

caee500e · Yaxun (Sam) Liu · GitHub · 8aecc1dc · caee500e
Unverified Commit caee500e authored May 04, 2020 by Yaxun (Sam) Liu Committed by GitHub May 04, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 0 deletions

src/targets/gpu/CMakeLists.txt src/targets/gpu/CMakeLists.txt +5 -0

No files found.
--- a/src/targets/gpu/CMakeLists.txt
+++ b/src/targets/gpu/CMakeLists.txt
@@ -70,6 +70,11 @@ set_target_properties(migraphx_device PROPERTIES EXPORT_NAME device)
 rocm_set_soversion(migraphx_device ${MIGRAPHX_SO_VERSION})
 rocm_clang_tidy_check(migraphx_device)
 target_compile_options(migraphx_device PRIVATE -std=c++17 -Xclang -fallow-half-arguments-and-returns)
+check_cxx_compiler_flag("--cuda-host-only -fhip-lambda-host-device -x hip" HAS_HIP_LAMBDA_HOST_DEVICE)
+if(HAS_HIP_LAMBDA_HOST_DEVICE)
+  message(STATUS "Enable -fhip-lambda-host-device")
+  target_compile_options(migraphx_device PRIVATE -fhip-lambda-host-device)
+endif()
 target_link_libraries(migraphx_device migraphx hip::device -Wno-invalid-command-line-argument -amdgpu-target=gfx803 -amdgpu-target=gfx900 -amdgpu-target=gfx906)
 target_include_directories(migraphx_device PUBLIC $<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}/include>)
 target_include_directories(migraphx_device PRIVATE $<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}/device/include>)