IntenseFinance/nosqlbench
3
0
Fork 0
mirror of https://github.com/nosqlbench/nosqlbench.git synced 2025-02-25 18:55:28 -06:00
Code Issues Packages Projects Releases Wiki Activity

fix buffer bug

Browse Source
This commit is contained in:
Jonathan Shook 2023-10-26 18:08:21 -05:00
parent cc5997c1dc
commit 92d62ae88c
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
virtdata-lib-hdf5/src/main/java/io/nosqlbench/virtdata/library/ivecfvec/FVecReader.java
View File

@ -83,11 +83,11 @@ public class FVecReader implements LongFunction<float[]> {
ByteBuffer record = this.bb.get(recpos,buf).order(ByteOrder.LITTLE_ENDIAN); ByteBuffer record = this.bb.get(recpos,buf).order(ByteOrder.LITTLE_ENDIAN);
int recdim = record.getInt(); int recdim = record.getInt();
if(recdim!=dimensions) { if(recdim!=dimensions) {
throw new RuntimeException("dimensions are not uniform for ivec file '" + this.path.toString() + "', found dim " + recdim + " at record " + value); throw new RuntimeException("dimensions are not uniform for fvec file '" + this.path.toString() + "', found dim " + recdim + " at record " + value);
} }
float[] data = new float[recdim]; float[] data = new float[recdim];
for (int i = 0; i < dimensions; i++) { for (int i = 0; i < dimensions; i++) {
data[i]=bb.getFloat(); data[i]=record.getFloat();
} }
return data; return data;
} }

2
virtdata-lib-hdf5/src/main/java/io/nosqlbench/virtdata/library/ivecfvec/IVecReader.java
View File

@ -99,7 +99,7 @@ public class IVecReader implements LongFunction<int[]> {
} }
int[] data = new int[recdim]; int[] data = new int[recdim];
for (int i = 0; i < dimensions; i++) { for (int i = 0; i < dimensions; i++) {
data[i]=Integer.reverseBytes(bb.getInt()); data[i]=Integer.reverseBytes(record.getInt());
} }
return data; return data;
} }