CMakeLists.txt

cmake_minimum_required(VERSION VERSION 3.21..3.26)
project(largemm)

# Find CUDA package
find_package(CUDA REQUIRED)

# Set CUDA architectures
set(CUDA_ARCHITECTURES "75" CACHE STRING "CUDA architectures to compile for")

# Set C++ standard to C++11
set(CMAKE_CXX_STANDARD 11)

# Set CUDA standard to 11
set(CMAKE_CUDA_STANDARD 11)
set(CMAKE_CUDA_STANDARD_REQUIRED ON)

# Add CUDA flags
set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -arch=sm_${CUDA_ARCHITECTURES}")

# Include directories
include_directories(src)

# # Add CUDA source files
# set(CUDA_SOURCE_FILES
#     src/largemm_base.cu
#     src/largemm_main.cu
#     src/largemm_test.cu
#     src/largemm_v1_4_streams.cu
# )

# Glob CUDA source files
file(GLOB CUDA_SOURCE_FILES
    src/*.cu
)

# Add header files
# set(HEADER_FILES
#     src/largemm-base.h
#     src/largemm-check.h
#     src/largemm-test.h
#     src/largemm-v1-4-streams.h
# )

# Glob header files
file(GLOB HEADER_FILES
    src/*.h
)

# Add executable target
cuda_add_executable(largemm
    ${CUDA_SOURCE_FILES}
    ${HEADER_FILES}
)

# Link CUBLAS library
target_link_libraries(largemm
    ${CUDA_CUBLAS_LIBRARIES}
)

# Install target
install(TARGETS largemm
        RUNTIME DESTINATION ${CMAKE_SOURCE_DIR}/bin
        LIBRARY DESTINATION ${CMAKE_SOURCE_DIR}/lib
        ARCHIVE DESTINATION ${CMAKE_SOURCE_DIR}/lib
)

# Install header files
install(FILES ${HEADER_FILES} DESTINATION ${CMAKE_SOURCE_DIR}/include)