ICODE-ADC/lammps
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update Kokkos library in LAMMPS to v4.6.0

Browse Source
This commit is contained in:
Stan Moore
2025-03-28 15:29:14 -06:00
parent 48893236ec
commit b7b9a4a599
384 changed files with 13243 additions and 9477 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
lib/kokkos/cmake/Modules/FindTPLCUDA.cmake
View File

@ -11,9 +11,16 @@ if(KOKKOS_CXX_HOST_COMPILER_ID STREQUAL NVHPC AND CMAKE_VERSION VERSION_LESS "3.
message(FATAL_ERROR "Using NVHPC as host compiler requires at least CMake 3.20.1")
endif()
set(TPL_CUDA_LIBRARIES "")
if(KOKKOS_ENABLE_COMPILE_AS_CMAKE_LANGUAGE)
set(TPL_CUDA_LIBRARIES CUDA::cuda_driver)
else()
set(TPL_CUDA_LIBRARIES CUDA::cuda_driver CUDA::cudart)
endif()
if(CMAKE_VERSION VERSION_GREATER_EQUAL "3.17.0")
find_package(CUDAToolkit REQUIRED)
kokkos_create_imported_tpl(CUDA INTERFACE LINK_LIBRARIES CUDA::cuda_driver CUDA::cudart)
kokkos_create_imported_tpl(CUDA INTERFACE LINK_LIBRARIES ${TPL_CUDA_LIBRARIES})
kokkos_export_cmake_tpl(CUDAToolkit REQUIRED)
else()
include(${CMAKE_CURRENT_LIST_DIR}/CudaToolkit.cmake)
@ -33,8 +40,8 @@ else()
endif()
include(FindPackageHandleStandardArgs)
find_package_handle_standard_args(TPLCUDA ${DEFAULT_MSG} FOUND_CUDART FOUND_CUDA_DRIVER)
find_package_handle_standard_args(TPLCUDA ${DEFAULT_MSG} FOUND_CUDA_DRIVER FOUND_CUDART)
if(FOUND_CUDA_DRIVER AND FOUND_CUDART)
kokkos_create_imported_tpl(CUDA INTERFACE LINK_LIBRARIES CUDA::cuda_driver CUDA::cudart)
kokkos_create_imported_tpl(CUDA INTERFACE LINK_LIBRARIES ${TPL_CUDA_LIBRARIES})
endif()
endif()