Skip to content

Commit 7496e6a

Browse files
committed
Set correctly number of bits for query and index
1 parent c4c2218 commit 7496e6a

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

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

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -132,8 +132,6 @@ public QuantizationResult scalarQuantize(float[] vector, byte[] destination, byt
132132
int assignment = Math.round((xi - a) / step);
133133
sumQuery += assignment;
134134
destination[h] = (byte) assignment;
135-
assert assignment >= 0 && assignment < points
136-
: "assignment: " + assignment + ", a: " + a + ", b: " + b + ", step: " + step + ", xi: " + xi;
137135
}
138136
return new QuantizationResult(
139137
intervalScratch[0],

server/src/main/java/org/elasticsearch/index/codec/vectors/es910/ES910BinaryQuantizedVectorsFormat.java

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -101,8 +101,8 @@ public class ES910BinaryQuantizedVectorsFormat extends FlatVectorsFormat {
101101
private static final DirectIOLucene99FlatVectorsFormat rawVectorFormat = new DirectIOLucene99FlatVectorsFormat(
102102
FlatVectorScorerUtil.getLucene99FlatVectorsScorer()
103103
);
104-
public static final byte DEFAULT_INDEX_BITS = (byte) 1;
105-
public static final byte DEFAULT_QUERY_BITS = (byte) 4;
104+
private static byte DEFAULT_INDEX_BITS = (byte) 1;
105+
private static byte DEFAULT_QUERY_BITS = (byte) 4;
106106

107107
private final ES910BinaryFlatVectorsScorer scorer;
108108

@@ -118,6 +118,10 @@ public ES910BinaryQuantizedVectorsFormat(byte indexBits, byte queryBits) {
118118
super(NAME);
119119
this.indexBits = indexBits;
120120
this.queryBits = queryBits;
121+
// Set the default bits for index and query vectors. I know, I know, this is a hack, but we
122+
// don't have the possibility of doing a PerFieldMapperCodec yet on KnnSearcher
123+
DEFAULT_QUERY_BITS = queryBits;
124+
DEFAULT_INDEX_BITS = indexBits;
121125
this.scorer = new ES910BinaryFlatVectorsScorer(FlatVectorScorerUtil.getLucene99FlatVectorsScorer(), queryBits);
122126
}
123127

0 commit comments

Comments
 (0)