Skip to navigation

LuceneAcceleratedHNSWScalarQuantizedVectorsWriter

View as Markdown

Java package: com.nvidia.cuvs.lucene

public class LuceneAcceleratedHNSWScalarQuantizedVectorsWriter extends KnnVectorsWriter

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

Public Members

LuceneAcceleratedHNSWScalarQuantizedVectorsWriter

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

Initializes LuceneAcceleratedHNSWScalarQuantizedVectorsWriter

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/LuceneAcceleratedHNSWScalarQuantizedVectorsWriter.java:90

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/LuceneAcceleratedHNSWScalarQuantizedVectorsWriter.java:138

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/LuceneAcceleratedHNSWScalarQuantizedVectorsWriter.java:242

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/LuceneAcceleratedHNSWScalarQuantizedVectorsWriter.java:326

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/LuceneAcceleratedHNSWScalarQuantizedVectorsWriter.java:357

close

@Override public void close() throws IOException

Closes the resources.

Source: java/cuvs-lucene/src/main/java/com/nvidia/cuvs/lucene/LuceneAcceleratedHNSWScalarQuantizedVectorsWriter.java:377

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/LuceneAcceleratedHNSWScalarQuantizedVectorsWriter.java:386

Source: java/cuvs-lucene/src/main/java/com/nvidia/cuvs/lucene/LuceneAcceleratedHNSWScalarQuantizedVectorsWriter.java:56