We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 8cbc095 commit f41d7ecCopy full SHA for f41d7ec
1 file changed
src/index/cagra/cagra.cu
@@ -22,13 +22,7 @@ static constexpr uint32_t cuda_concurrent_size = 16;
22
23
namespace knowhere {
24
25
-static std::shared_ptr<ThreadPool>
26
-GlobalThreadPoolRaft() {
27
- static std::shared_ptr<ThreadPool> pool = std::make_shared<ThreadPool>(cuda_concurrent_size);
28
- return pool;
29
-}
30
-
31
KNOWHERE_REGISTER_GLOBAL(GPU_RAFT_CAGRA, [](const Object& object) {
32
- return Index<IndexNodeThreadPoolWrapper>::Create(std::make_unique<CagraIndexNode>(object), GlobalThreadPoolRaft());
+ return Index<IndexNodeThreadPoolWrapper>::Create(std::make_unique<CagraIndexNode>(object), cuda_concurrent_size);
33
});
34
} // namespace knowhere
0 commit comments