add CUDA-aware persistent and send/recv

This commit is contained in:
Carl Pearson
2021-06-02 16:50:54 -04:00
parent d29de778d2
commit 16de9c455b
4 changed files with 340 additions and 10 deletions

View File

@@ -127,3 +127,19 @@ if (MPI_FOUND AND CMAKE_CUDA_COMPILER)
set_cxx_options(one-sided-gpu)
set_cxx_standard(one-sided-gpu)
endif()
if (MPI_FOUND AND CMAKE_CUDA_COMPILER)
add_executable( persistent-gpu persistent_gpu.cpp)
target_link_libraries(persistent-gpu MPI::MPI_CXX)
target_link_libraries(persistent-gpu CUDA::cudart)
set_cxx_options( persistent-gpu)
set_cxx_standard( persistent-gpu)
endif()
if (MPI_FOUND AND CMAKE_CUDA_COMPILER)
add_executable(send-recv-gpu send_recv_gpu.cpp)
target_link_libraries(send-recv-gpu MPI::MPI_CXX)
target_link_libraries(send-recv-gpu CUDA::cudart)
set_cxx_options(send-recv-gpu)
set_cxx_standard(send-recv-gpu)
endif()