🥷 fix loop bound type

di-dimmasik · Sep 29, 2023 · ed8babb · ed8babb
1 parent 47c43b5
commit ed8babb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/jvector-twenty/src/main/java/io/github/jbellis/jvector/vector/SimdOps.java b/jvector-twenty/src/main/java/io/github/jbellis/jvector/vector/SimdOps.java
@@ -555,7 +555,7 @@ static float assembleAndSum512(float[] data, int dataBase, byte[] baseOffsets) {
         int[] convOffsets = scratchInt512.get();
         FloatVector sum = FloatVector.zero(FloatVector.SPECIES_512);
         int i = 0;
-        int limit = FloatVector.SPECIES_128.loopBound(baseOffsets.length);
+        int limit = ByteVector.SPECIES_128.loopBound(baseOffsets.length);
 
         for (; i < limit; i += ByteVector.SPECIES_128.length()) {
             var scale = IntVector.zero(IntVector.SPECIES_512).addIndex(1).add(i).mul(dataBase);
@@ -583,7 +583,7 @@ static float assembleAndSum256(float[] data, int dataBase, byte[] baseOffsets) {
         int[] convOffsets = scratchInt256.get();
         FloatVector sum = FloatVector.zero(FloatVector.SPECIES_256);
         int i = 0;
-        int limit = FloatVector.SPECIES_64.loopBound(baseOffsets.length);
+        int limit = ByteVector.SPECIES_64.loopBound(baseOffsets.length);
 
         for (; i < limit; i += ByteVector.SPECIES_64.length()) {
             var scale = IntVector.zero(IntVector.SPECIES_256).addIndex(1).add(i).mul(dataBase);