tweak GPU.cmake so that hip_get_devices compiles and links (29d66772) · Commits · 郑智淋 / lammps

cmake/Modules/Packages/GPU.cmake

+4 −0

Original line number	Diff line number	Diff line
		@@ -338,6 +338,10 @@ elseif(GPU_API STREQUAL "HIP")
		target_include_directories(hip_get_devices PRIVATE ${HIP_ROOT_DIR}/include)
		target_include_directories(hip_get_devices PRIVATE ${CUDA_INCLUDE_DIRS})
		target_link_libraries(hip_get_devices PRIVATE ${CUDA_LIBRARIES} ${CUDA_CUDA_LIBRARY})
		elseif(HIP_PLATFORM STREQUAL "hcc")
		target_compile_definitions(hip_get_devices PRIVATE -D__HIP_PLATFORM_HCC__)
		target_include_directories(hip_get_devices PRIVATE ${HIP_ROOT_DIR}/include)
		target_include_directories(hip_get_devices PRIVATE ${HIP_ROOT_DIR}/../hsa/include)
		endif()

		target_link_libraries(lammps PRIVATE gpu)