Pulkitg64 commented on code in PR #16383:
URL: https://github.com/apache/lucene/pull/16383#discussion_r3573139445
##########
lucene/core/src/java/org/apache/lucene/codecs/lucene99/Lucene99FlatVectorsWriter.java:
##########
@@ -347,6 +379,26 @@ private static DocsWithFieldSet writeVectorData(
return docsWithField;
}
+ /**
+ * Writes the vector values to the output and returns a set of documents
that contains vectors.
+ */
+ private static DocsWithFieldSet writeFloat16VectorData(
+ IndexOutput output, Float16VectorValues float16VectorValues) throws
IOException {
+ DocsWithFieldSet docsWithField = new DocsWithFieldSet();
+ ByteBuffer buffer =
+ ByteBuffer.allocate(float16VectorValues.dimension() *
VectorEncoding.FLOAT16.byteSize)
+ .order(ByteOrder.LITTLE_ENDIAN);
+ KnnVectorValues.DocIndexIterator iter = float16VectorValues.iterator();
+ for (int docV = iter.nextDoc(); docV != NO_MORE_DOCS; docV =
iter.nextDoc()) {
+ // write vector
+ short[] value = float16VectorValues.vectorValue(iter.index());
+ buffer.asShortBuffer().put(value);
Review Comment:
Make sense. Fixed in next revision.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
