From 82dca79b7dd9b54f5a90a17b372eb0e3962daa64 Mon Sep 17 00:00:00 2001 From: usamoi Date: Fri, 10 Oct 2025 16:59:32 +0800 Subject: [PATCH] feat: AIO batchmode Signed-off-by: usamoi --- src/index/vchordg/scanners/default.rs | 4 ++-- src/index/vchordrq/scanners/default.rs | 4 ++-- src/index/vchordrq/scanners/maxsim.rs | 4 ++-- 3 files changed, 6 insertions(+), 6 deletions(-) diff --git a/src/index/vchordg/scanners/default.rs b/src/index/vchordg/scanners/default.rs index b0945bf4..4c309849 100644 --- a/src/index/vchordg/scanners/default.rs +++ b/src/index/vchordg/scanners/default.rs @@ -108,8 +108,8 @@ impl SearchBuilder for DefaultBuilder { let Some(vector) = vector else { return Box::new(std::iter::empty()) as Box>; }; - let vertex_hints = Hints::default().full(true); - let vector_hints = Hints::default().full(true); + let vertex_hints = Hints::default().full(true).batch(true); + let vector_hints = Hints::default().full(true).batch(true); let make_vertex_plain_prefetcher = MakePlainPrefetcher { index }; let make_vertex_simple_prefetcher = MakeSimplePrefetcher { index }; let make_vertex_stream_prefetcher = MakeStreamPrefetcher { diff --git a/src/index/vchordrq/scanners/default.rs b/src/index/vchordrq/scanners/default.rs index 388396f2..f8969644 100644 --- a/src/index/vchordrq/scanners/default.rs +++ b/src/index/vchordrq/scanners/default.rs @@ -111,8 +111,8 @@ impl SearchBuilder for DefaultBuilder { let Some(vector) = vector else { return Box::new(std::iter::empty()) as Box>; }; - let search_hints = Hints::default().full(true); - let rerank_hints = Hints::default().full(false); + let search_hints = Hints::default().full(true).batch(false); + let rerank_hints = Hints::default().full(false).batch(true); let make_h1_plain_prefetcher = MakeH1PlainPrefetcher { index }; let make_h0_plain_prefetcher = MakeH0PlainPrefetcher { index }; let make_h0_simple_prefetcher = MakeH0SimplePrefetcher { index }; diff --git a/src/index/vchordrq/scanners/maxsim.rs b/src/index/vchordrq/scanners/maxsim.rs index 934a2e94..ad77adb9 100644 --- a/src/index/vchordrq/scanners/maxsim.rs +++ b/src/index/vchordrq/scanners/maxsim.rs @@ -103,8 +103,8 @@ impl SearchBuilder for MaxsimBuilder { pgrx::error!("maxsim search with rerank_in_table is not supported"); } assert!(matches!(opfamily.distance_kind(), DistanceKind::Dot)); - let search_hints = Hints::default().full(true); - let rerank_hints = Hints::default().full(false); + let search_hints = Hints::default().full(true).batch(false); + let rerank_hints = Hints::default().full(false).batch(true); let make_h1_plain_prefetcher = MakeH1PlainPrefetcher { index }; let make_h0_plain_prefetcher = MakeH0PlainPrefetcher { index }; let make_h0_simple_prefetcher = MakeH0SimplePrefetcher { index };