fix:miss wait thread tasks finish in diskann. (#357)

Signed-off-by: cqy123456 <qianya.cheng@zilliz.com>
zilliztech · Jan 22, 2024 · b14a4d6 · b14a4d6
1 parent 433de97
commit b14a4d6
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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);