
# 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)
get_built_tool_path(
  TOOL_enum_parser_bin
  TOOL_enum_parser_dependency
  tools/enum_parser/enum_parser
  enum_parser
)

add_library(catboost-cuda-gpu_data)
target_cuda_sources(catboost-cuda-gpu_data PRIVATE
  ${CMAKE_SOURCE_DIR}/catboost/cuda/gpu_data/kernel/split.cu
  ${CMAKE_SOURCE_DIR}/catboost/cuda/gpu_data/kernel/query_helper.cu
  ${CMAKE_SOURCE_DIR}/catboost/cuda/gpu_data/kernel/binarize.cu
)
target_link_libraries(catboost-cuda-gpu_data PUBLIC
  contrib-libs-cxxsupp
  yutil
  library-cpp-grid_creator
  cpp-threading-local_executor
  catboost-cuda-cuda_lib
  catboost-cuda-cuda_util
  catboost-cuda-data
  catboost-cuda-ctrs
  private-libs-ctr_description
  catboost-libs-data
  catboost-libs-helpers
  private-libs-quantization
  private-libs-quantized_pool
  tools-enum_parser-enum_serialization_runtime
  CUDA::toolkit
)
target_sources(catboost-cuda-gpu_data PRIVATE
  ${CMAKE_SOURCE_DIR}/catboost/cuda/gpu_data/batch_binarized_ctr_calcer.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/gpu_data/cat_features_dataset.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/gpu_data/compressed_index.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/gpu_data/compressed_index_builder.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/gpu_data/dataset_helpers.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/gpu_data/feature_parallel_dataset.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/gpu_data/doc_parallel_dataset_builder.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/gpu_data/doc_parallel_dataset.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/gpu_data/feature_parallel_dataset_builder.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/gpu_data/oblivious_tree_bin_builder.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/gpu_data/pinned_memory_estimation.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/gpu_data/samples_grouping.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/gpu_data/samples_grouping_gpu.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/gpu_data/querywise_helper.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/gpu_data/bootstrap.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/gpu_data/non_zero_filter.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/gpu_data/ctr_helper.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/gpu_data/feature_layout.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/gpu_data/gpu_binarization_helpers.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/gpu_data/estimated_features_calcer.cpp
)
generate_enum_serilization(catboost-cuda-gpu_data
  ${CMAKE_SOURCE_DIR}/catboost/cuda/gpu_data/grid_policy.h
  INCLUDE_HEADERS
  catboost/cuda/gpu_data/grid_policy.h
)
target_cuda_cflags(catboost-cuda-gpu_data)
target_cuda_flags(catboost-cuda-gpu_data
  -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(catboost-cuda-gpu_data.global catboost-cuda-gpu_data)
target_link_libraries(catboost-cuda-gpu_data.global PUBLIC
  contrib-libs-cxxsupp
  yutil
  library-cpp-grid_creator
  cpp-threading-local_executor
  catboost-cuda-cuda_lib
  catboost-cuda-cuda_util
  catboost-cuda-data
  catboost-cuda-ctrs
  private-libs-ctr_description
  catboost-libs-data
  catboost-libs-helpers
  private-libs-quantization
  private-libs-quantized_pool
  tools-enum_parser-enum_serialization_runtime
  CUDA::toolkit
)
target_sources(catboost-cuda-gpu_data.global PRIVATE
  ${CMAKE_SOURCE_DIR}/catboost/cuda/gpu_data/kernels.cpp
  ${CMAKE_SOURCE_DIR}/catboost/cuda/gpu_data/splitter.cpp
)
