From d3baa414d2d686f0b9bc7b2f2be44f45937a5277 Mon Sep 17 00:00:00 2001 From: Vsevak Date: Wed, 17 Jun 2020 23:33:16 +0300 Subject: [PATCH] Fix cmake for GPU ROCm 3.5 backend --- cmake/Modules/Packages/GPU.cmake | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/cmake/Modules/Packages/GPU.cmake b/cmake/Modules/Packages/GPU.cmake index d3a73100c0..bc890d6691 100644 --- a/cmake/Modules/Packages/GPU.cmake +++ b/cmake/Modules/Packages/GPU.cmake @@ -269,7 +269,7 @@ elseif(GPU_API STREQUAL "HIP") configure_file(${CU_FILE} ${CU_CPP_FILE} COPYONLY) add_custom_command(OUTPUT ${CUBIN_FILE} - VERBATIM COMMAND ${HIP_HIPCC_EXECUTABLE} --genco -t="${HIP_ARCH}" -f=\"-O3 -ffast-math -DUSE_HIP -D_${GPU_PREC_SETTING} -I${LAMMPS_LIB_SOURCE_DIR}/gpu\" -o ${CUBIN_FILE} ${CU_CPP_FILE} + VERBATIM COMMAND ${HIP_HIPCC_EXECUTABLE} --genco --offload-arch=${HIP_ARCH} -O3 -ffast-math -DUSE_HIP -D_${GPU_PREC_SETTING} -I${LAMMPS_LIB_SOURCE_DIR}/gpu -o ${CUBIN_FILE} ${CU_CPP_FILE} DEPENDS ${CU_CPP_FILE} COMMENT "Generating ${CU_NAME}.cubin") elseif(HIP_PLATFORM STREQUAL "nvcc") @@ -298,6 +298,8 @@ elseif(GPU_API STREQUAL "HIP") # add hipCUB target_include_directories(gpu PRIVATE ${HIP_ROOT_DIR}/../include) target_compile_definitions(gpu PRIVATE -DUSE_HIP_DEVICE_SORT) + find_package(rocprim REQUIRED CONFIG PATHS "/opt/rocm/rocprim") + find_package(hipcub REQUIRED CONFIG PATHS "/opt/rocm/hipcub") if(HIP_PLATFORM STREQUAL "nvcc") find_package(CUB)