Skip to content

Commit 73a10ad

Browse files
committed
Reduce quantization optimization steps at ivf query time
1 parent 136442d commit 73a10ad

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

server/src/main/java/org/elasticsearch/index/codec/vectors/DefaultIVFVectorsReader.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,7 @@
3333
import static org.apache.lucene.index.VectorSimilarityFunction.MAXIMUM_INNER_PRODUCT;
3434
import static org.elasticsearch.index.codec.vectors.BQSpaceUtils.transposeHalfByte;
3535
import static org.elasticsearch.index.codec.vectors.BQVectorUtils.discretize;
36+
import static org.elasticsearch.index.codec.vectors.OptimizedScalarQuantizer.DEFAULT_LAMBDA;
3637
import static org.elasticsearch.simdvec.ES91OSQVectorsScorer.BULK_SIZE;
3738

3839
/**
@@ -211,7 +212,7 @@ private static class MemorySegmentPostingsVisitor implements PostingVisitor {
211212
quantizedQueryScratch = new byte[QUERY_BITS * discretizedDimensions / 8];
212213
quantizedByteLength = discretizedDimensions / 8 + (Float.BYTES * 3) + Short.BYTES;
213214
quantizedVectorByteSize = (discretizedDimensions / 8);
214-
quantizer = new OptimizedScalarQuantizer(fieldInfo.getVectorSimilarityFunction());
215+
quantizer = new OptimizedScalarQuantizer(fieldInfo.getVectorSimilarityFunction(), DEFAULT_LAMBDA, 2);
215216
osqVectorsScorer = ESVectorUtil.getES91OSQVectorsScorer(indexInput, fieldInfo.getVectorDimension());
216217
}
217218

0 commit comments

Comments
 (0)