Skip to navigation

LuceneAcceleratedHNSWBinaryQuantizedVectorsWriter

View as Markdown

Java package: com.nvidia.cuvs.lucene

public class LuceneAcceleratedHNSWBinaryQuantizedVectorsWriter extends KnnVectorsWriter

This class extends upon the KnnVectorsWriter to enable the creation of GPU-based accelerated vector search indexes.

Public Members

LuceneAcceleratedHNSWBinaryQuantizedVectorsWriter

public LuceneAcceleratedHNSWBinaryQuantizedVectorsWriter( SegmentWriteState state, AcceleratedHNSWParams acceleratedHNSWParams, FlatVectorsWriter flatVectorsWriter) throws IOException

Initializes LuceneAcceleratedHNSWBinaryQuantizedVectorsWriter

Parameters

NameDescription
stateinstance of the org.apache.lucene.index.SegmentWriteState
acceleratedHNSWParamsAn instance of AcceleratedHNSWParams
flatVectorsWriterinstance of the org.apache.lucene.codecs.hnsw.FlatVectorsWriter

Throws

TypeDescription
IOExceptionIOException

Source: java/cuvs-lucene/src/main/java/com/nvidia/cuvs/lucene/LuceneAcceleratedHNSWBinaryQuantizedVectorsWriter.java:80

addField

@Override public KnnFieldVectorsWriter<?> addField(FieldInfo fieldInfo) throws IOException

Add new field for indexing.

Source: java/cuvs-lucene/src/main/java/com/nvidia/cuvs/lucene/LuceneAcceleratedHNSWBinaryQuantizedVectorsWriter.java:129

flush

@Override public void flush(int maxDoc, DocMap sortMap) throws IOException

Build the indexes and writes it to the disk.

Source: java/cuvs-lucene/src/main/java/com/nvidia/cuvs/lucene/LuceneAcceleratedHNSWBinaryQuantizedVectorsWriter.java:233

mergeOneField

@Override public void mergeOneField(FieldInfo fieldInfo, MergeState mergeState) throws IOException

Write field for merging.

Source: java/cuvs-lucene/src/main/java/com/nvidia/cuvs/lucene/LuceneAcceleratedHNSWBinaryQuantizedVectorsWriter.java:319

finish

@Override public void finish() throws IOException

Called once at the end before close.

Source: java/cuvs-lucene/src/main/java/com/nvidia/cuvs/lucene/LuceneAcceleratedHNSWBinaryQuantizedVectorsWriter.java:350

close

@Override public void close() throws IOException

Closes the resources.

Source: java/cuvs-lucene/src/main/java/com/nvidia/cuvs/lucene/LuceneAcceleratedHNSWBinaryQuantizedVectorsWriter.java:370

ramBytesUsed

@Override public long ramBytesUsed()

Returns the memory usage of this object in bytes.

Source: java/cuvs-lucene/src/main/java/com/nvidia/cuvs/lucene/LuceneAcceleratedHNSWBinaryQuantizedVectorsWriter.java:379

Source: java/cuvs-lucene/src/main/java/com/nvidia/cuvs/lucene/LuceneAcceleratedHNSWBinaryQuantizedVectorsWriter.java:57