CMakeLists.txt

cmake_minimum_required(VERSION "3.1.0")
include(cmake/custom_functions.cmake)
project(root)

find_package(Torch REQUIRED)

option(USE_CUDA  "Build with CUDA")

include(GNUInstallDirs)

find_library(CATBOOST catboost)
find_path(CATBOOST_INCLUDES_API catboost_api.h contrib/catboost)
include_directories(${CATBOOST_INCLUDES_API})
include_directories(contrib/catboost/)
# WTF this not work...
#IF (WIN32)
##    add_library(catboost STATIC IMPORTED)
##    set_property(TARGET catboost PROPERTY IMPORTED_LOCATION "libcatboost.lib")
#ELSE()
#    IF(APPLE)
#        add_library(catboost SHARED IMPORTED)
#        set_property(TARGET catboost PROPERTY IMPORTED_LOCATION ${CMAKE_CURRENT_SOURCE_DIR}/libcatboost.dylib)
#    ELSE()
#        add_library(catboost SHARED IMPORTED)
#        set_property(TARGET catboost PROPERTY IMPORTED_LOCATION "libcatboost.so")
#    ENDIF())
#ENDIF()

include_directories(contrib/json)

add_subdirectory(contrib/googletest)
include_directories(contrib/googletest/include contrib/googletest)
find_library(GTEST gtest NO_DEFAULT_PATH)
message("gtest path: ${GTEST}")
add_subdirectory(contrib/pybind11)
enable_testing()

set(CMAKE_POSITION_INDEPENDENT_CODE ON)

if(USE_CUDA)
    enable_language(CUDA)
    add_definitions(-DCUDA)
    set(CMAKE_CUDA_STANDARD 14)
    set(CMAKE_CUDA_STANDARD_REQUIRED ON)
#    SET(CMAKE_CUDA_FLAGS "–cudart static")
    set(CMAKE_CUDA_COMPILE_SEPARABLE_COMPILATION ON)
    include_directories(cub)

    set(CUDA_NVCC_FLAGS "--expt-relaxed-constexpr
                            -gencode arch=compute_30,code=compute_30
                            -gencode arch=compute_35,code=sm_35
                            -gencode arch=compute_50,code=compute_50
                            -gencode arch=compute_52,code=sm_52
                            -gencode arch=compute_60,code=compute_60
                            -gencode arch=compute_61,code=compute_61
                            -gencode arch=compute_61,code=sm_61
                            -gencode arch=compute_70,code=sm_70
                            -gencode arch=compute_70,code=compute_70
                            --ptxas-options=-v")
endif()

add_subdirectory(cpp)