From b14a4d620151980e0242754f9a51eeae39b3b660 Mon Sep 17 00:00:00 2001 From: cqy123456 <39671710+cqy123456@users.noreply.github.com> Date: Mon, 22 Jan 2024 18:08:49 +0800 Subject: [PATCH] fix:miss wait thread tasks finish in diskann. (#357) Signed-off-by: cqy123456 --- thirdparty/DiskANN/src/partition_and_pq.cpp | 3 +++ 1 file changed, 3 insertions(+) diff --git a/thirdparty/DiskANN/src/partition_and_pq.cpp b/thirdparty/DiskANN/src/partition_and_pq.cpp index c8078a22e..47bf0dd6f 100644 --- a/thirdparty/DiskANN/src/partition_and_pq.cpp +++ b/thirdparty/DiskANN/src/partition_and_pq.cpp @@ -530,6 +530,9 @@ int generate_pq_data_from_pivots(const std::string data_file, } })); } + for (auto &future : futures) { + future.wait(); + } futures.clear(); futures.reserve(num_pq_chunks);