diff --git a/src/index/ivf_raft/ivf_raft.cu b/src/index/ivf_raft/ivf_raft.cu index b7120f3b8..024cc660c 100644 --- a/src/index/ivf_raft/ivf_raft.cu +++ b/src/index/ivf_raft/ivf_raft.cu @@ -18,24 +18,30 @@ #include "knowhere/factory.h" #include "knowhere/index_node_thread_pool_wrapper.h" +constexpr uint32_t cuda_concurrent_size = 16; + namespace knowhere { KNOWHERE_REGISTER_GLOBAL(GPU_RAFT_IVF_FLAT, [](const Object& object) { return Index::Create( - std::make_unique>(object)); + std::make_unique>(object), + std::make_shared(cuda_concurrent_size)); }); KNOWHERE_REGISTER_GLOBAL(GPU_RAFT_IVF_PQ, [](const Object& object) { return Index::Create( - std::make_unique>(object)); + std::make_unique>(object), + std::make_shared(cuda_concurrent_size)); }); KNOWHERE_REGISTER_GLOBAL(GPU_IVF_FLAT, [](const Object& object) { return Index::Create( - std::make_unique>(object)); + std::make_unique>(object), + std::make_shared(cuda_concurrent_size)); }); KNOWHERE_REGISTER_GLOBAL(GPU_IVF_PQ, [](const Object& object) { return Index::Create( - std::make_unique>(object)); + std::make_unique>(object), + std::make_shared(cuda_concurrent_size)); }); } // namespace knowhere