Skip to content

Commit f41d7ec

Browse files
committed
Apply Threadpool changes
1 parent 8cbc095 commit f41d7ec

1 file changed

Lines changed: 1 addition & 7 deletions

File tree

src/index/cagra/cagra.cu

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -22,13 +22,7 @@ static constexpr uint32_t cuda_concurrent_size = 16;
2222

2323
namespace knowhere {
2424

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-
3125
KNOWHERE_REGISTER_GLOBAL(GPU_RAFT_CAGRA, [](const Object& object) {
32-
return Index<IndexNodeThreadPoolWrapper>::Create(std::make_unique<CagraIndexNode>(object), GlobalThreadPoolRaft());
26+
return Index<IndexNodeThreadPoolWrapper>::Create(std::make_unique<CagraIndexNode>(object), cuda_concurrent_size);
3327
});
3428
} // namespace knowhere

0 commit comments

Comments
 (0)