
# This file was generated by the build system used internally in Yandex and called "ya"
# (https://github.com/yandex/yatool).
#
# Configuration files of ya build system are usually named ya.make. If ya.make file is present
# at the root of the repository, then this repository supports ya build. 
# 
# If the repository supports both CMake and ya build configurations, please modify both of them.
#
# If only CMake build configuration is supported then modify only CMake files and note that only
# simple modifications are allowed like adding source-files to targets or adding simple properties
# like target_include_directories. These modifications will be ported to original ya.make files
# by maintainers. Any complex modifications which can't be easily ported back to the ya build 
# system may be rejected.

find_package(CUDAToolkit REQUIRED)
add_library(models)


target_cuda_sources(models PRIVATE
  ${PROJECT_SOURCE_DIR}/catboost/cuda/models/kernel/add_model_value.cu
)

target_link_libraries(models PUBLIC
  contrib-libs-linux-headers
  contrib-libs-cxxsupp
  yutil
  catboost-cuda-cuda_lib
  catboost-cuda-cuda_util
  catboost-cuda-gpu_data
  CUDA::toolkit
)

target_sources(models PRIVATE
  ${PROJECT_SOURCE_DIR}/catboost/cuda/models/add_oblivious_tree_model_doc_parallel.cpp
  ${PROJECT_SOURCE_DIR}/catboost/cuda/models/oblivious_model.cpp
  ${PROJECT_SOURCE_DIR}/catboost/cuda/models/region_model.cpp
  ${PROJECT_SOURCE_DIR}/catboost/cuda/models/non_summetric_tree.cpp
  ${PROJECT_SOURCE_DIR}/catboost/cuda/models/model_converter.cpp
  ${PROJECT_SOURCE_DIR}/catboost/cuda/models/compact_model.cpp
)



target_cuda_cflags(models)

target_cuda_flags(models
  -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
  -gencode=arch=compute_80,code=sm_80
  -gencode=arch=compute_86,code=sm_86
  -gencode=arch=compute_89,code=sm_89
  -gencode=arch=compute_90,code=sm_90
)


add_global_library_for(models.global
  models)


target_link_libraries(models.global PUBLIC
  contrib-libs-linux-headers
  contrib-libs-cxxsupp
  yutil
  catboost-cuda-cuda_lib
  catboost-cuda-cuda_util
  catboost-cuda-gpu_data
  CUDA::toolkit
)

target_sources(models.global PRIVATE
  ${PROJECT_SOURCE_DIR}/catboost/cuda/models/add_bin_values.cpp
  ${PROJECT_SOURCE_DIR}/catboost/cuda/models/add_region_doc_parallel.cpp
  ${PROJECT_SOURCE_DIR}/catboost/cuda/models/add_non_symmetric_tree_doc_parallel.cpp
)






