From 2b0e17ede0a113f5694d3407431eef394ca60663 Mon Sep 17 00:00:00 2001 From: Kerollmops Date: Thu, 6 Feb 2025 15:28:10 +0100 Subject: [PATCH] Make sure arroy is using the rayon thread-pool --- crates/milli/src/update/new/indexer/mod.rs | 17 ++++++++++------- 1 file changed, 10 insertions(+), 7 deletions(-) diff --git a/crates/milli/src/update/new/indexer/mod.rs b/crates/milli/src/update/new/indexer/mod.rs index 8b98bfba3..bb8292750 100644 --- a/crates/milli/src/update/new/indexer/mod.rs +++ b/crates/milli/src/update/new/indexer/mod.rs @@ -191,13 +191,16 @@ where indexing_context.progress.update_progress(IndexingStep::WritingEmbeddingsToDatabase); - build_vectors( - index, - wtxn, - index_embeddings, - &mut arroy_writers, - &indexing_context.must_stop_processing, - )?; + pool.install(|| { + build_vectors( + index, + wtxn, + index_embeddings, + &mut arroy_writers, + &indexing_context.must_stop_processing, + ) + }) + .unwrap()?; post_processing::post_process( indexing_context,