Module org.elasticsearch.simdvec

Package org.elasticsearch.simdvec.internal

Class Int7uOSQVectorScorerSupplier

java.lang.Object

org.elasticsearch.simdvec.internal.Int7uOSQVectorScorerSupplier

All Implemented Interfaces:

org.apache.lucene.util.hnsw.RandomVectorScorerSupplier

Direct Known Subclasses:

Int7uOSQVectorScorerSupplier.DotProductSupplier, Int7uOSQVectorScorerSupplier.EuclideanSupplier, Int7uOSQVectorScorerSupplier.MaxInnerProductSupplier

public abstract sealed class Int7uOSQVectorScorerSupplier
extends Object
implements org.apache.lucene.util.hnsw.RandomVectorScorerSupplier
permits Int7uOSQVectorScorerSupplier.DotProductSupplier, Int7uOSQVectorScorerSupplier.EuclideanSupplier, Int7uOSQVectorScorerSupplier.MaxInnerProductSupplier

Int7 OSQ scorer supplier backed by MemorySegmentAccessInput storage.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static final class	Int7uOSQVectorScorerSupplier.DotProductSupplier
static final class	Int7uOSQVectorScorerSupplier.EuclideanSupplier
static final class	Int7uOSQVectorScorerSupplier.MaxInnerProductSupplier
protected static final record	Int7uOSQVectorScorerSupplier.QueryContext

Field Summary

Fields

Modifier and Type	Field	Description
protected final int	dims
protected final org.apache.lucene.store.IndexInput	input
protected final int	maxOrd
protected final org.apache.lucene.util.quantization.QuantizedByteVectorValues	values
protected final int	vectorPitch

Method Summary

Modifier and Type	Method	Description
protected abstract float	applyCorrections(float rawScore, MemorySegment correctionsSlice, Int7uOSQVectorScorerSupplier.QueryContext query)
protected abstract float	applyCorrectionsBulk(MemorySegment scores, MemorySegment addrs, int numNodes, Int7uOSQVectorScorerSupplier.QueryContext query)
protected final float	bulkScoreFromOrds(Int7uOSQVectorScorerSupplier.QueryContext query, int[] ordinals, float[] scores, int numNodes)
protected final void	checkOrdinal(int ord)
protected Int7uOSQVectorScorerSupplier.QueryContext	createQueryContext(int ord)
org.apache.lucene.util.quantization.QuantizedByteVectorValues	get()
protected final float	scoreFromOrds(Int7uOSQVectorScorerSupplier.QueryContext query, int secondOrd)
org.apache.lucene.util.hnsw.UpdateableRandomVectorScorer	scorer()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface org.apache.lucene.util.hnsw.RandomVectorScorerSupplier

copy

Field Details

input

protected final org.apache.lucene.store.IndexInput input

values

protected final org.apache.lucene.util.quantization.QuantizedByteVectorValues values

dims

protected final int dims

maxOrd

protected final int maxOrd

vectorPitch

protected final int vectorPitch

Method Details

applyCorrections

protected abstract float applyCorrections(float rawScore,
 MemorySegment correctionsSlice,
 Int7uOSQVectorScorerSupplier.QueryContext query)
                                   throws IOException

Throws:

IOException

applyCorrectionsBulk

protected abstract float applyCorrectionsBulk(MemorySegment scores,
 MemorySegment addrs,
 int numNodes,
 Int7uOSQVectorScorerSupplier.QueryContext query)
                                       throws IOException

Throws:

IOException

createQueryContext

protected Int7uOSQVectorScorerSupplier.QueryContext createQueryContext(int ord)
                                                                throws IOException

Throws:

IOException

checkOrdinal

protected final void checkOrdinal(int ord)

scoreFromOrds

protected final float scoreFromOrds(Int7uOSQVectorScorerSupplier.QueryContext query,
 int secondOrd)
                             throws IOException

Throws:

IOException

bulkScoreFromOrds

protected final float bulkScoreFromOrds(Int7uOSQVectorScorerSupplier.QueryContext query,
 int[] ordinals,
 float[] scores,
 int numNodes)
                                 throws IOException

Throws:

IOException

scorer

public org.apache.lucene.util.hnsw.UpdateableRandomVectorScorer scorer()

Specified by:

scorer in interface org.apache.lucene.util.hnsw.RandomVectorScorerSupplier

get

public org.apache.lucene.util.quantization.QuantizedByteVectorValues get()