110 строки
4.2 KiB
CMake
110 строки
4.2 KiB
CMake
|
# Copyright (c) Microsoft Corporation. All rights reserved.
|
||
|
|
# Licensed under the MIT license.
|
||
|
|
|
||
|
set(CMAKE_CXX_STANDARD 17)
|
||
|
set(CMAKE_COMPILE_WARNING_AS_ERROR ON)
|
||
|
|
|
||
|
|
|
||
|
|
add_executable(fvecs_to_bin fvecs_to_bin.cpp)
|
||
|
|
|
||
|
add_executable(fvecs_to_bvecs fvecs_to_bvecs.cpp)
|
||
|
|
|
||
|
add_executable(rand_data_gen rand_data_gen.cpp)
|
||
|
target_link_libraries(rand_data_gen ${PROJECT_NAME} Boost::program_options)
|
||
|
|
|
||
|
add_executable(float_bin_to_int8 float_bin_to_int8.cpp)
|
||
|
|
|
||
|
|
add_executable(ivecs_to_bin ivecs_to_bin.cpp)
|
||
|
|
|
||
|
|
add_executable(count_bfs_levels count_bfs_levels.cpp)
|
||
|
|
target_link_libraries(count_bfs_levels ${PROJECT_NAME} Boost::program_options)
|
||
|
|
|
||
|
add_executable(tsv_to_bin tsv_to_bin.cpp)
|
||
|
|
|
||
|
|
add_executable(bin_to_tsv bin_to_tsv.cpp)
|
||
|
|
|
||
|
|
add_executable(int8_to_float int8_to_float.cpp)
|
||
|
|
target_link_libraries(int8_to_float ${PROJECT_NAME})
|
||
|
|
|
||
|
|
add_executable(int8_to_float_scale int8_to_float_scale.cpp)
|
||
|
|
target_link_libraries(int8_to_float_scale ${PROJECT_NAME})
|
||
|
|
|
||
|
|
add_executable(uint8_to_float uint8_to_float.cpp)
|
||
|
|
target_link_libraries(uint8_to_float ${PROJECT_NAME})
|
||
|
|
|
||
|
|
add_executable(uint32_to_uint8 uint32_to_uint8.cpp)
|
||
|
|
target_link_libraries(uint32_to_uint8 ${PROJECT_NAME})
|
||
|
|
|
||
|
|
add_executable(vector_analysis vector_analysis.cpp)
|
||
|
|
target_link_libraries(vector_analysis ${PROJECT_NAME} ${DISKANN_TOOLS_TCMALLOC_LINK_OPTIONS})
|
||
|
|
|
||
|
|
add_executable(gen_random_slice gen_random_slice.cpp)
|
||
|
|
target_link_libraries(gen_random_slice ${PROJECT_NAME} ${DISKANN_TOOLS_TCMALLOC_LINK_OPTIONS})
|
||
|
|
|
||
|
|
add_executable(simulate_aggregate_recall simulate_aggregate_recall.cpp)
|
||
|
|
|
||
|
|
add_executable(calculate_recall calculate_recall.cpp)
|
||
|
|
target_link_libraries(calculate_recall ${PROJECT_NAME} ${DISKANN_ASYNC_LIB} ${DISKANN_TOOLS_TCMALLOC_LINK_OPTIONS})
|
||
|
|
|
||
|
|
# Compute ground truth thing outside of DiskANN main source that depends on MKL.
|
||
|
|
add_executable(compute_groundtruth compute_groundtruth.cpp)
|
||
|
|
target_include_directories(compute_groundtruth PRIVATE ${DISKANN_MKL_INCLUDE_DIRECTORIES})
|
||
|
|
target_link_libraries(compute_groundtruth ${PROJECT_NAME} ${DISKANN_MKL_LINK_LIBRARIES} ${DISKANN_ASYNC_LIB} Boost::program_options)
|
||
|
|
|
||
|
|
add_executable(compute_groundtruth_for_filters compute_groundtruth_for_filters.cpp)
|
||
|
|
target_include_directories(compute_groundtruth_for_filters PRIVATE ${DISKANN_MKL_INCLUDE_DIRECTORIES})
|
||
|
|
target_link_libraries(compute_groundtruth_for_filters ${PROJECT_NAME} ${DISKANN_MKL_LINK_LIBRARIES} ${DISKANN_ASYNC_LIB} Boost::program_options)
|
||
|
|
|
||
|
|
||
|
|
add_executable(generate_pq generate_pq.cpp)
|
||
|
|
target_link_libraries(generate_pq ${PROJECT_NAME} ${DISKANN_TOOLS_TCMALLOC_LINK_OPTIONS})
|
||
|
|
|
||
|
|
|
||
|
|
add_executable(partition_data partition_data.cpp)
|
||
|
|
target_link_libraries(partition_data ${PROJECT_NAME} ${DISKANN_TOOLS_TCMALLOC_LINK_OPTIONS})
|
||
|
|
|
||
|
|
add_executable(partition_with_ram_budget partition_with_ram_budget.cpp)
|
||
|
|
target_link_libraries(partition_with_ram_budget ${PROJECT_NAME} ${DISKANN_TOOLS_TCMALLOC_LINK_OPTIONS})
|
||
|
|
|
||
|
|
add_executable(merge_shards merge_shards.cpp)
|
||
|
|
target_link_libraries(merge_shards ${PROJECT_NAME} ${DISKANN_TOOLS_TCMALLOC_LINK_OPTIONS} ${DISKANN_ASYNC_LIB})
|
||
|
|
|
||
|
|
add_executable(create_disk_layout create_disk_layout.cpp)
|
||
|
|
target_link_libraries(create_disk_layout ${PROJECT_NAME} ${DISKANN_ASYNC_LIB} ${DISKANN_TOOLS_TCMALLOC_LINK_OPTIONS})
|
||
|
|
|
||
|
|
add_executable(generate_synthetic_labels generate_synthetic_labels.cpp)
|
||
|
|
target_link_libraries(generate_synthetic_labels ${PROJECT_NAME} Boost::program_options)
|
||
|
|
|
||
|
|
add_executable(stats_label_data stats_label_data.cpp)
|
||
|
|
target_link_libraries(stats_label_data ${PROJECT_NAME} Boost::program_options)
|
||
|
|
||
|
|
if (NOT MSVC)
|
||
|
|
include(GNUInstallDirs)
|
||
|
|
install(TARGETS fvecs_to_bin
|
||
|
|
fvecs_to_bvecs
|
||
|
|
rand_data_gen
|
||
|
|
float_bin_to_int8
|
||
|
|
ivecs_to_bin
|
||
|
|
count_bfs_levels
|
||
|
|
tsv_to_bin
|
||
|
|
bin_to_tsv
|
||
|
|
int8_to_float
|
||
|
|
int8_to_float_scale
|
||
|
|
uint8_to_float
|
||
|
|
uint32_to_uint8
|
||
|
|
vector_analysis
|
||
|
|
gen_random_slice
|
||
|
|
simulate_aggregate_recall
|
||
|
|
calculate_recall
|
||
|
|
compute_groundtruth
|
||
|
|
compute_groundtruth_for_filters
|
||
|
|
generate_pq
|
||
|
|
partition_data
|
||
|
|
partition_with_ram_budget
|
||
|
|
merge_shards
|
||
|
|
create_disk_layout
|
||
|
|
generate_synthetic_labels
|
||
|
|
stats_label_data
|
||
|
|
RUNTIME
|
||
|
|
)
|
||
|
|
endif()
|