
# This file was generated by the build system used internally in the Yandex monorepo.
# Only simple modifications are allowed (adding source-files to targets, 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 ported back to the
# original buildsystem will not be accepted.


find_package(CUDAToolkit REQUIRED)
add_subdirectory(bin)
add_subdirectory(ut)

add_library(catboost-cuda-targets)
target_cuda_sources(catboost-cuda-targets PRIVATE
  ${CMAKE_SOURCE_DIR}/catboost/cuda/targets/kernel/dcg.cu
  ${CMAKE_SOURCE_DIR}/catboost/cuda/targets/kernel/multilogit.cu
  ${CMAKE_SOURCE_DIR}/catboost/cuda/targets/kernel/pair_logit.cu
  ${CMAKE_SOURCE_DIR}/catboost/cuda/targets/kernel/pfound_f.cu
  ${CMAKE_SOURCE_DIR}/catboost/cuda/targets/kernel/pointwise_targets.cu
  ${CMAKE_SOURCE_DIR}/catboost/cuda/targets/kernel/query_cross_entropy.cu
  ${CMAKE_SOURCE_DIR}/catboost/cuda/targets/kernel/query_rmse.cu
  ${CMAKE_SOURCE_DIR}/catboost/cuda/targets/kernel/query_softmax.cu
  ${CMAKE_SOURCE_DIR}/catboost/cuda/targets/kernel/yeti_rank_pointwise.cu
)
target_link_libraries(catboost-cuda-targets PUBLIC
  contrib-libs-cxxsupp
  yutil
  catboost-cuda-cuda_lib
  catboost-cuda-cuda_util
  catboost-cuda-gpu_data
  catboost-libs-data
  catboost-libs-helpers
  catboost-libs-logging
  catboost-libs-metrics
  private-libs-options
  cpp-threading-local_executor
  CUDA::toolkit
)
target_sources(catboost-cuda-targets PRIVATE
  ${CMAKE_SOURCE_DIR}/catboost/cuda/targets/auc.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/targets/combination_targets_impl.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/targets/gpu_metrics.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/targets/multiclass_targets.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/targets/pair_logit_pairwise.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/targets/pfound_f.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/targets/pointwise_target_impl.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/targets/query_cross_entropy.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/targets/querywise_targets_impl.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/targets/target_func.cpp
)
target_cuda_cflags(catboost-cuda-targets)
target_cuda_flags(catboost-cuda-targets
  -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
)

add_global_library_for(catboost-cuda-targets.global catboost-cuda-targets)
target_link_libraries(catboost-cuda-targets.global PUBLIC
  contrib-libs-cxxsupp
  yutil
  catboost-cuda-cuda_lib
  catboost-cuda-cuda_util
  catboost-cuda-gpu_data
  catboost-libs-data
  catboost-libs-helpers
  catboost-libs-logging
  catboost-libs-metrics
  private-libs-options
  cpp-threading-local_executor
  CUDA::toolkit
)
target_sources(catboost-cuda-targets.global PRIVATE
  ${CMAKE_SOURCE_DIR}/catboost/cuda/targets/dcg.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/targets/kernel.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/targets/multiclass_kernels.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/targets/query_cross_entropy_kernels.cpp
)
