cmake: updates for gt stencil2d

This commit is contained in:
Bryce Allen
2023-03-25 10:53:50 -07:00
parent 02ab50ab60
commit 7967991bd9

View File

@@ -6,30 +6,35 @@ project(mpi-daxpy-test)
# add dependencies
include(cmake/CPM.cmake)
CPMFindPackage(NAME gtensor
GITHUB_REPOSITORY wdmapp/gtensor
GIT_TAG "main"
GITHUB_REPOSITORY bd4/gtensor
GIT_TAG "pr/view-contiguous"
OPTIONS "GTENSOR_ENABLE_BLAS ON")
set(MPI_CXX_SKIP_MPICXX ON)
find_package(MPI REQUIRED)
if ("${GTENSOR_DEVICE}" STREQUAL "cuda")
enable_language(CUDA)
endif()
add_executable(mpi_daxpy_gt)
target_sources(mpi_daxpy_gt PRIVATE mpi_daxpy_gt.cc)
target_link_libraries(mpi_daxpy_gt gtensor::gtensor)
target_link_libraries(mpi_daxpy_gt gtensor::blas)
target_link_libraries(mpi_daxpy_gt MPI::MPI_CXX)
target_link_libraries(mpi_daxpy_gt PRIVATE gtensor::gtensor)
target_link_libraries(mpi_daxpy_gt PRIVATE gtensor::blas)
target_link_libraries(mpi_daxpy_gt PRIVATE MPI::MPI_CXX)
add_executable(mpi_stencil_gt)
target_sources(mpi_stencil_gt PRIVATE mpi_stencil_gt.cc)
target_link_libraries(mpi_stencil_gt gtensor::gtensor)
target_link_libraries(mpi_stencil_gt MPI::MPI_CXX)
target_link_libraries(mpi_stencil_gt PRIVATE gtensor::gtensor)
target_link_libraries(mpi_stencil_gt PRIVATE MPI::MPI_CXX)
add_executable(mpi_stencil2d_gt)
target_sources(mpi_stencil2d_gt PRIVATE mpi_stencil2d_gt.cc)
target_link_libraries(mpi_stencil2d_gt gtensor::gtensor)
target_link_libraries(mpi_stencil2d_gt MPI::MPI_CXX)
target_link_libraries(mpi_stencil2d_gt PRIVATE gtensor::gtensor)
target_link_libraries(mpi_stencil2d_gt PRIVATE MPI::MPI_CXX)
target_compile_features(mpi_stencil2d_gt PRIVATE cxx_std_17)
if ("${GTENSOR_DEVICE}" STREQUAL "cuda")
enable_language(CUDA)
set_source_files_properties(mpi_daxpy_gt.cc
TARGET_DIRECTORY mpi_daxpy_gt
PROPERTIES LANGUAGE CUDA)
@@ -37,7 +42,7 @@ if ("${GTENSOR_DEVICE}" STREQUAL "cuda")
TARGET_DIRECTORY mpi_stencil_gt
PROPERTIES LANGUAGE CUDA)
set_source_files_properties(mpi_stencil2d_gt.cc
TARGET_DIRECTORY mpi_stencil_gt
TARGET_DIRECTORY mpi_stencil2d_gt
PROPERTIES LANGUAGE CUDA)
else()
set_source_files_properties(mpi_daxpy_gt.cc
@@ -47,7 +52,7 @@ else()
TARGET_DIRECTORY mpi_stencil_gt
PROPERTIES LANGUAGE CXX)
set_source_files_properties(mpi_stencil2d_gt.cc
TARGET_DIRECTORY mpi_stencil_gt
TARGET_DIRECTORY mpi_stencil2d_gt
PROPERTIES LANGUAGE CXX)
endif()