Skip to content

Commit a6101e5

Browse files
authored
fix: allocate much memory for residual vectors than needed (#3446)
Signed-off-by: BubbleCal <bubble-cal@outlook.com>
1 parent 7687558 commit a6101e5

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

rust/lance-index/src/vector/residual.rs

+2-1
Original file line numberDiff line numberDiff line change
@@ -91,9 +91,10 @@ where
9191
let c = &centroids_slice[part_id * dimension..(part_id + 1) * dimension];
9292
iter::zip(vector, c).map(|(v, cent)| *v - *cent)
9393
})
94-
.exact_size(vectors.len() * dimension)
94+
.exact_size(vectors.len())
9595
.collect::<Vec<_>>();
9696
let residual_arr = PrimitiveArray::<T>::from_iter_values(residuals);
97+
debug_assert_eq!(residual_arr.len(), vectors.len());
9798
Ok(FixedSizeListArray::try_new_from_values(
9899
residual_arr,
99100
dimension as i32,

0 commit comments

Comments
 (0)