From 0561c4b0cba46f632230e88b911551882ea3ee39 Mon Sep 17 00:00:00 2001 From: "xj.lin" Date: Fri, 30 Aug 2019 15:14:23 +0800 Subject: [PATCH] MS-442 fix merge error Former-commit-id: 26cc2b6dc6b7b1c0a9d1b3d687bc3bf06097080f --- cpp/src/core/src/knowhere/index/vector_index/gpu_ivf.cpp | 1 + cpp/src/core/test/test_nsg/CMakeLists.txt | 2 ++ 2 files changed, 3 insertions(+) diff --git a/cpp/src/core/src/knowhere/index/vector_index/gpu_ivf.cpp b/cpp/src/core/src/knowhere/index/vector_index/gpu_ivf.cpp index 01c4908511..c06c58c40c 100644 --- a/cpp/src/core/src/knowhere/index/vector_index/gpu_ivf.cpp +++ b/cpp/src/core/src/knowhere/index/vector_index/gpu_ivf.cpp @@ -123,6 +123,7 @@ void GPUIVF::search_impl(int64_t n, int64_t *labels, const Config &cfg) { if (auto device_index = std::static_pointer_cast(index_)) { + // todo: allocate search memory auto nprobe = cfg.get_with_default("nprobe", size_t(1)); std::lock_guard lk(mutex_); diff --git a/cpp/src/core/test/test_nsg/CMakeLists.txt b/cpp/src/core/test/test_nsg/CMakeLists.txt index bd2db8072c..0ec275d9ee 100644 --- a/cpp/src/core/test/test_nsg/CMakeLists.txt +++ b/cpp/src/core/test/test_nsg/CMakeLists.txt @@ -27,6 +27,8 @@ aux_source_directory(${CMAKE_SOURCE_DIR}/src/knowhere/index/vector_index/nsg nsg set(interface_src ${CMAKE_SOURCE_DIR}/src/knowhere/index/vector_index/ivf.cpp ${CMAKE_SOURCE_DIR}/src/knowhere/index/vector_index/gpu_ivf.cpp + ${CMAKE_SOURCE_DIR}/src/knowhere/index/vector_index/cloner.cpp + ${CMAKE_SOURCE_DIR}/src/knowhere/index/vector_index/idmap.cpp ${CMAKE_SOURCE_DIR}/src/knowhere/index/vector_index/nsg_index.cpp ${CMAKE_SOURCE_DIR}/src/knowhere/adapter/structure.cpp ${CMAKE_SOURCE_DIR}/src/knowhere/common/exception.cpp