Fix DiskANN Build Percentage (#391)

Signed-off-by: Patrick Weizhi Xu <[email protected]>
zilliztech · Feb 19, 2024 · da19c51 · da19c51
1 parent a3cbc25
commit da19c51
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/thirdparty/DiskANN/src/partition_and_pq.cpp b/thirdparty/DiskANN/src/partition_and_pq.cpp
@@ -350,7 +350,8 @@ int generate_pq_pivots(const float *passed_train_data, size_t num_train,
                     chunk_size * sizeof(float));
       }
       uint32_t cur_chunk_done = num_chunk_done.fetch_add(1);
-      if (cur_chunk_done % num_chunk_step == 0 || cur_chunk_done == num_pq_chunks) {
+      if (num_chunk_step == 0 || cur_chunk_done % num_chunk_step == 0 ||
+          cur_chunk_done == num_pq_chunks) {
         LOG_KNOWHERE_INFO_ << "Genereated pivots for " << cur_chunk_done
                            << " chunks out of " << num_pq_chunks;
       }