Fix missing cuda library dependencies

This commit is contained in:
Ye Luo 2021-06-17 16:28:56 -05:00
parent 5e6974aeba
commit 54f46aa7f3
3 changed files with 9 additions and 1 deletions

View File

@ -215,6 +215,8 @@ if(QE_ENABLE_CUDA OR QE_ENABLE_PROFILE_NVTX)
set_target_properties(CUDA::cublas PROPERTIES INTERFACE_LINK_LIBRARIES "${CUDA_FLAG}lib=cublas")
add_library(CUDA::cusolver INTERFACE IMPORTED)
set_target_properties(CUDA::cusolver PROPERTIES INTERFACE_LINK_LIBRARIES "${CUDA_FLAG}lib=cusolver")
add_library(CUDA::curand INTERFACE IMPORTED)
set_target_properties(CUDA::curand PROPERTIES INTERFACE_LINK_LIBRARIES "${CUDA_FLAG}lib=curand")
else()
find_package(CUDAToolkit REQUIRED)
endif()

View File

@ -42,7 +42,8 @@ if(QE_ENABLE_CUDA)
endif()
target_link_libraries(qe_lax
PRIVATE
CUDA::cusolver)
CUDA::cusolver
CUDA::cublas)
set_property(TARGET qe_lax APPEND
PROPERTY
COMPILE_DEFINITIONS __USE_CUSOLVER)

View File

@ -148,6 +148,11 @@ target_link_libraries(qe_modules
qe_fox
qe_utilx)
if(QE_ENABLE_CUDA)
target_link_libraries(qe_modules
PRIVATE
CUDA::curand)
endif()
###########################################################
qe_install_targets(qe_modules)