diff --git a/libs/native/src/test/java/org/elasticsearch/nativeaccess/jdk/JDKVectorLibraryInt4Tests.java b/libs/native/src/test/java/org/elasticsearch/nativeaccess/jdk/JDKVectorLibraryInt4Tests.java index 525c82db46ff0..d1f68b97338ad 100644 --- a/libs/native/src/test/java/org/elasticsearch/nativeaccess/jdk/JDKVectorLibraryInt4Tests.java +++ b/libs/native/src/test/java/org/elasticsearch/nativeaccess/jdk/JDKVectorLibraryInt4Tests.java @@ -126,7 +126,8 @@ public void testInt4Bulk() { float[] expectedScores = new float[numVecs]; scalarSimilarityBulk(unpackedValues[queryOrd], packedValues, expectedScores); - var nativeQuerySeg = MemorySegment.ofArray(unpackedValues[queryOrd]); + var nativeQuerySeg = arena.allocate(dims); + MemorySegment.copy(unpackedValues[queryOrd], 0, nativeQuerySeg, ValueLayout.JAVA_BYTE, 0L, dims); var bulkScoresSeg = arena.allocate((long) numVecs * Float.BYTES); similarityBulk(packedSegment, nativeQuerySeg, packedLen, numVecs, bulkScoresSeg); assertScoresEquals(expectedScores, bulkScoresSeg); @@ -162,7 +163,8 @@ public void testInt4BulkWithOffsets() { float[] expectedScores = new float[numVecs]; scalarSimilarityBulkWithOffsets(unpackedValues[queryOrd], packedValues, offsets, expectedScores); - var nativeQuerySeg = MemorySegment.ofArray(unpackedValues[queryOrd]); + var nativeQuerySeg = arena.allocate(dims); + MemorySegment.copy(unpackedValues[queryOrd], 0, nativeQuerySeg, ValueLayout.JAVA_BYTE, 0L, dims); var bulkScoresSeg = arena.allocate((long) numVecs * Float.BYTES); similarityBulkWithOffsets(packedSegment, nativeQuerySeg, packedLen, packedLen, offsetsSegment, numVecs, bulkScoresSeg); @@ -195,7 +197,8 @@ public void testInt4BulkWithOffsetsAndPitch() { float[] expectedScores = new float[numVecs]; scalarSimilarityBulkWithOffsets(unpackedValues[queryOrd], packedValues, offsets, expectedScores); - var nativeQuerySeg = MemorySegment.ofArray(unpackedValues[queryOrd]); + var nativeQuerySeg = arena.allocate(dims); + MemorySegment.copy(unpackedValues[queryOrd], 0, nativeQuerySeg, ValueLayout.JAVA_BYTE, 0L, dims); var bulkScoresSeg = arena.allocate((long) numVecs * Float.BYTES); similarityBulkWithOffsets(packedSegment, nativeQuerySeg, packedLen, pitch, offsetsSegment, numVecs, bulkScoresSeg);