
# 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(ut)

add_library(catboost-cuda-cuda_util)
target_cuda_sources(catboost-cuda-cuda_util PRIVATE
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/kernel/bootstrap.cu
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/kernel/compression.cu
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/kernel/dot_product.cu
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/kernel/fill.cu
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/kernel/filter.cu
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/kernel/mvs.cu
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/kernel/operators.cu
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/kernel/partitions.cu
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/kernel/random.cu
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/kernel/reduce.cu
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/kernel/reorder_one_bit.cu
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/kernel/scan.cu
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/kernel/segmented_scan.cu
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/kernel/segmented_sort.cu
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/kernel/sort/bool_uchar.cu
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/kernel/sort/bool_ui32.cu
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/kernel/sort/float_uchar.cu
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/kernel/sort/float_ui16.cu
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/kernel/sort/float_ui32.cu
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/kernel/sort/float_uint2.cu
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/kernel/sort/ui32_uchar.cu
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/kernel/sort/ui32_ui16.cu
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/kernel/sort/ui32_ui32.cu
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/kernel/sort/ui64_uchar.cu
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/kernel/sort/ui64_ui32.cu
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/kernel/transform.cu
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/kernel/update_part_props.cu
)
target_link_libraries(catboost-cuda-cuda_util PUBLIC
  contrib-libs-linux-headers
  contrib-libs-cxxsupp
  yutil
  catboost-libs-helpers
  catboost-cuda-cuda_lib
  libs-nvidia-cub
  CUDA::toolkit
)
target_cuda_cflags(catboost-cuda-cuda_util)
target_cuda_flags(catboost-cuda-cuda_util
  -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-cuda_util.global catboost-cuda-cuda_util)
target_link_libraries(catboost-cuda-cuda_util.global PUBLIC
  contrib-libs-linux-headers
  contrib-libs-cxxsupp
  yutil
  catboost-libs-helpers
  catboost-cuda-cuda_lib
  libs-nvidia-cub
  CUDA::toolkit
)
target_sources(catboost-cuda-cuda_util.global PRIVATE
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/bootstrap.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/compression_helpers_gpu.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/dot_product.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/fill.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/filter.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/gpu_random.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/helpers.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/partitions.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/partitions_reduce.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/reduce.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/reorder_bins.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/scan.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/segmented_scan.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/segmented_sort.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/sort.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/cuda_util/transform.cpp
)
