CMakeLists.txt

cmake_minimum_required(VERSION 3.18)
project(BlockStructuredSolvers LANGUAGES CXX CUDA)

set(CMAKE_CXX_STANDARD 17)
set(CMAKE_CUDA_STANDARD 17)

# Find required packages
find_package(CUDAToolkit REQUIRED)
find_package(BLAS REQUIRED)
find_package(LAPACK REQUIRED)

# Include directories
include_directories(include)

# Source files
set(CPU_SOURCES
    src/cpu_solver.cpp
    src/block_tridiag_data.cpp
)

set(CUDA_SOURCES
    src/cuda_solver.cu
    src/cuda_batched_solver.cu
)

# Create the library
add_library(block_structured_solvers ${CPU_SOURCES} ${CUDA_SOURCES})

# Link libraries
target_link_libraries(block_structured_solvers 
    CUDA::cublas 
    CUDA::cusolver 
    CUDA::cudart
    ${BLAS_LIBRARIES}
    ${LAPACK_LIBRARIES}
)

# Define CUDA_AVAILABLE for conditional compilation
target_compile_definitions(block_structured_solvers PUBLIC CUDA_AVAILABLE)

# Set CUDA properties
set_property(TARGET block_structured_solvers PROPERTY CUDA_ARCHITECTURES 70 75 80 86)

# Examples
add_executable(example_cpu examples/example_cpu.cpp)
target_link_libraries(example_cpu block_structured_solvers)

add_executable(example_cuda examples/example_cuda.cpp)
target_link_libraries(example_cuda block_structured_solvers)

add_executable(batched_cuda_example examples/batched_cuda_example.cu)
target_link_libraries(batched_cuda_example block_structured_solvers)
set_property(TARGET batched_cuda_example PROPERTY CUDA_ARCHITECTURES 70 75 80 86)

# Tests
add_executable(test_solver tests/test_solver.cpp)
target_link_libraries(test_solver block_structured_solvers)

# Benchmarks
add_executable(benchmark benchmarks/benchmark.cpp)
target_link_libraries(benchmark block_structured_solvers)

# Tests
enable_testing()
add_test(NAME solver_tests COMMAND test_solver)