Skip to content

Commit c2a8068

Browse files
committed
Fixed species check for cosine in scoreFunction
1 parent 58ef33f commit c2a8068

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

jvector-twenty/src/main/java/io/github/jbellis/jvector/vector/PanamaVectorUtilSupport.java

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1909,7 +1909,7 @@ float pqScoreCosine_512(VectorFloat<?>[] codebooks, int[][] subvectorSizesAndOff
19091909
int length1 = centroidIndex1 * centroidLength;
19101910
int length2 = centroidIndex2 * centroidLength;
19111911

1912-
if (centroidLength == FloatVector.SPECIES_512.length()) {
1912+
if (centroidLength == FloatVector.SPECIES_PREFERRED.length()) {
19131913
FloatVector a = fromVectorFloat(FloatVector.SPECIES_PREFERRED, codebooks[m], length1);
19141914
FloatVector b = fromVectorFloat(FloatVector.SPECIES_PREFERRED, codebooks[m], length2);
19151915
vSum = a.fma(b, vSum);
@@ -2416,7 +2416,7 @@ float pqScoreCosine_512(VectorFloat<?>[] codebooks, int[][] subvectorSizesAndOff
24162416
int length1 = centroidIndex * centroidLength;
24172417
int length2 = subvectorSizesAndOffsets[m][1];
24182418

2419-
if (centroidLength == FloatVector.SPECIES_512.length()) {
2419+
if (centroidLength == FloatVector.SPECIES_PREFERRED.length()) {
24202420
FloatVector a = fromVectorFloat(FloatVector.SPECIES_PREFERRED, codebooks[m], length1);
24212421
FloatVector b = fromVectorFloat(FloatVector.SPECIES_PREFERRED, centeredQuery, length2);
24222422
vSum = a.fma(b, vSum);
@@ -2462,7 +2462,7 @@ float pqScoreCosine_256(VectorFloat<?>[] codebooks, int[][] subvectorSizesAndOff
24622462
int length1 = centroidIndex * centroidLength;
24632463
int length2 = subvectorSizesAndOffsets[m][1];
24642464

2465-
if (centroidLength == FloatVector.SPECIES_512.length()) {
2465+
if (centroidLength == FloatVector.SPECIES_256.length()) {
24662466
FloatVector a = fromVectorFloat(FloatVector.SPECIES_256, codebooks[m], length1);
24672467
FloatVector b = fromVectorFloat(FloatVector.SPECIES_256, centeredQuery, length2);
24682468
vSum = a.fma(b, vSum);
@@ -2508,7 +2508,7 @@ float pqScoreCosine_128(VectorFloat<?>[] codebooks, int[][] subvectorSizesAndOff
25082508
int length1 = centroidIndex * centroidLength;
25092509
int length2 = subvectorSizesAndOffsets[m][1];
25102510

2511-
if (centroidLength == FloatVector.SPECIES_512.length()) {
2511+
if (centroidLength == FloatVector.SPECIES_128.length()) {
25122512
FloatVector a = fromVectorFloat(FloatVector.SPECIES_128, codebooks[m], length1);
25132513
FloatVector b = fromVectorFloat(FloatVector.SPECIES_128, centeredQuery, length2);
25142514
vSum = a.fma(b, vSum);
@@ -2553,7 +2553,7 @@ float pqScoreCosine_64(VectorFloat<?>[] codebooks, int[][] subvectorSizesAndOffs
25532553
int length1 = centroidIndex * centroidLength;
25542554
int length2 = subvectorSizesAndOffsets[m][1];
25552555

2556-
if (centroidLength == FloatVector.SPECIES_512.length()) {
2556+
if (centroidLength == FloatVector.SPECIES_64.length()) {
25572557
FloatVector a = fromVectorFloat(FloatVector.SPECIES_64, codebooks[m], length1);
25582558
FloatVector b = fromVectorFloat(FloatVector.SPECIES_64, centeredQuery, length2);
25592559
vSum = a.fma(b, vSum);

0 commit comments

Comments
 (0)