CMakeLists.txt

cmake_minimum_required(VERSION 3.14)
SITE_NAME(HOSTNAME)
if(${HOSTNAME} STREQUAL "j63c09243.sqa.eu95")
set(CMAKE_CUDA_COMPILER /usr/local/cuda-10.1/bin/nvcc)
elseif(${HOSTNAME} STREQUAL "meepo4")
set(CMAKE_CUDA_COMPILER /usr/local/cuda-10.2/bin/nvcc)
elseif(${HOSTNAME} STREQUAL "meepo5")
set(CMAKE_CUDA_COMPILER /usr/local/cuda-11/bin/nvcc)
elseif(${HOSTNAME} STREQUAL "SH-IDC1-10-142-40-63")
set(CMAKE_CUDA_COMPILER /usr/local/cuda-11.7/bin/nvcc)
else()
message( STATUS "host is $ENV{HOSTNAME}")
message( WARNING "unknown host" )
endif()

project(coll_cache_library CXX CUDA)

set(CMAKE_BUILD_TYPE "Release")
# set(CMAKE_BUILD_TYPE "Debug")

if (CMAKE_BUILD_TYPE STREQUAL "Debug")
  set(COMMON_FLAGS " -O0 -g ")
endif()

if (CMAKE_BUILD_TYPE STREQUAL "Release")
  set(COMMON_FLAGS " -O3 -g ")
endif()

set(CMAKE_EXPORT_COMPILE_COMMANDS ON)

set(CMAKE_CXX_STANDARD 14)
# no-as-needed is necessary because gurobi95 is not directly used by coll_cache
set(CMAKE_CXX_FLAGS  "${CMAKE_CXX_FLAGS}  ${COMMON_FLAGS} -fPIC -Wall -fopenmp -march=native -D_GLIBCXX_USE_CXX11_ABI=0 -DUNIT_TEST -DPSTL_USE_PARALLEL_POLICIES=0 -Wl,--no-as-needed")
set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} ${COMMON_FLAGS} --compiler-options '-fopenmp -fPIC -D_GLIBCXX_USE_CXX11_ABI=0' -gencode=arch=compute_70,code=sm_70 -gencode=arch=compute_80,code=sm_80")
message(DEBUG "CMAKE_CXX_FLAGS: ${CMAKE_CXX_FLAGS}")
message(DEBUG "CMAKE_CUDA_FLAGS: ${CMAKE_CUDA_FLAGS}")
include_directories(
  ${PROJECT_SOURCE_DIR}/coll_cache_lib
  ${CMAKE_CUDA_TOOLKIT_INCLUDE_DIRECTORIES}
  ../3rdparty/CLI11/include
  $ENV{GUROBI_HOME}/include
)

# source scripts already adds necessary path to LD path
link_directories(
  # $ENV{GUROBI_HOME}/lib
  # $ENV{CONDA_PREFIX}/lib
  # /opt/intel/oneapi/tbb/latest/lib/intel64/gcc4.8
  /usr/local/cuda/lib64
)


file(GLOB_RECURSE COLL_CACHE_SRC
  coll_cache_lib/*.cc
  coll_cache_lib/*.cu)

# build samgraph library
add_library(coll_cache
  SHARED
  ${COLL_CACHE_SRC}
)
target_link_libraries(coll_cache
  PUBLIC
  gurobi100
  gurobi_c++
  tbb
  cuda
  cudart
  nvToolsExt)


# add_executable(
#   samgraph_cpp_no_train
#   main.cc
# )
# target_link_libraries(
#   samgraph_cpp_no_train
#   coll_cache
# )

# # for unit test
# 
# message(STATUS "Build with unittest")
# add_subdirectory(../3rdparty/googletest googletest)
# include(GoogleTest)
# enable_testing()
# include_directories(${gtest_SOURCE_DIR}/include ${gtest_SOURCE_DIR})
# include_directories(${PROJECT_SOURCE_DIR})
# file(GLOB_RECURSE TEST_SRC_FILES 
#   ${PROJECT_SOURCE_DIR}/unittest/*.cc
#   ${PROJECT_SOURCE_DIR}/unittest/*.cu
# )
# add_executable(runUnitTests ${TEST_SRC_FILES})
# target_link_libraries(runUnitTests gtest gtest_main)
# target_link_libraries(runUnitTests samgraph_lib)
# gtest_discover_tests(runUnitTests)