Skip to content

feat: AIO batchmode #353

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
usamoi wants to merge 1 commit into
base: main
Choose a base branch
from
Open

feat: AIO batchmode #353

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions src/index/vchordg/scanners/default.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -108,8 +108,8 @@ impl SearchBuilder for DefaultBuilder {
let Some(vector) = vector else {
return Box::new(std::iter::empty()) as Box<dyn Iterator<Item = (f32, [u16; 3], bool)>>;
};
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 {
Expand Down
4 changes: 2 additions & 2 deletions src/index/vchordrq/scanners/default.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -111,8 +111,8 @@ impl SearchBuilder for DefaultBuilder {
let Some(vector) = vector else {
return Box::new(std::iter::empty()) as Box<dyn Iterator<Item = (f32, [u16; 3], bool)>>;
};
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 };
Expand Down
4 changes: 2 additions & 2 deletions src/index/vchordrq/scanners/maxsim.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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 };
Expand Down
Loading