HnswQuery

Java package: com.nvidia.cuvs

1
public class HnswQuery

HnswQuery holds the query vectors to be used while invoking search on the HNSW index.

Thread Safety: Each HnswQuery instance should use its own CuVSResources object that is not shared with other threads. Sharing CuVSResources between threads can lead to memory allocation errors or JVM crashes.

Public Members

HnswQuery

1
private HnswQuery( HnswSearchParams hnswSearchParams, float[][] queryVectors, LongToIntFunction mapping, int topK, CuVSResources resources)

Constructs an instance of HnswQuery using queryVectors, mapping, and topK.

Parameters

Source: java/cuvs-java/src/main/java/com/nvidia/cuvs/HnswQuery.java:38

getHnswSearchParams

1
public HnswSearchParams getHnswSearchParams()

Gets the instance of HnswSearchParams.

Returns

the instance of HnswSearchParams

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

getQueryVectors

1
public float[][] getQueryVectors()

Gets the query vector 2D float array.

Returns

2D float array

Source: java/cuvs-java/src/main/java/com/nvidia/cuvs/HnswQuery.java:65

getMapping

1
public LongToIntFunction getMapping()

Gets the function mapping ordinals (neighbor IDs) to custom user IDs

Source: java/cuvs-java/src/main/java/com/nvidia/cuvs/HnswQuery.java:72

getTopK

1
public int getTopK()

Gets the topK value.

Returns

an integer

Source: java/cuvs-java/src/main/java/com/nvidia/cuvs/HnswQuery.java:81

getResources

1
public CuVSResources getResources()

Gets the CuVSResources instance for this query.

Returns

the CuVSResources instance

Source: java/cuvs-java/src/main/java/com/nvidia/cuvs/HnswQuery.java:90

Builder

1
public Builder(CuVSResources resources)

Constructor that requires CuVSResources.

Important: The provided CuVSResources instance should not be shared with other threads. Each thread performing searches should create its own CuVSResources instance to avoid memory allocation conflicts and potential JVM crashes.

Parameters

Source: java/cuvs-java/src/main/java/com/nvidia/cuvs/HnswQuery.java:125

withSearchParams

1
public Builder withSearchParams(HnswSearchParams hnswSearchParams)

Sets the instance of configured HnswSearchParams to be passed for search.

Parameters

Returns

an instance of this Builder

Source: java/cuvs-java/src/main/java/com/nvidia/cuvs/HnswQuery.java:136

withQueryVectors

1
public Builder withQueryVectors(float[][] queryVectors)

Registers the query vectors to be passed in the search call.

Parameters

Returns

an instance of this Builder

Source: java/cuvs-java/src/main/java/com/nvidia/cuvs/HnswQuery.java:147

withMapping

1
public Builder withMapping(LongToIntFunction mapping)

Sets the function used to map ordinals (neighbor IDs) to custom user IDs

Parameters

Returns

an instance of this Builder

Source: java/cuvs-java/src/main/java/com/nvidia/cuvs/HnswQuery.java:158

withTopK

1
public Builder withTopK(int topK)

Registers the topK value.

Parameters

Returns

an instance of this Builder

Source: java/cuvs-java/src/main/java/com/nvidia/cuvs/HnswQuery.java:169

build

1
public HnswQuery build()

Builds an instance of HnswQuery

Returns

an instance of HnswQuery

Source: java/cuvs-java/src/main/java/com/nvidia/cuvs/HnswQuery.java:179

Source: java/cuvs-java/src/main/java/com/nvidia/cuvs/HnswQuery.java:21