@@ -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