CuVSMatrix

Java package: com.nvidia.cuvs

1
public interface CuVSMatrix extends AutoCloseable

This represents a wrapper for a dataset to be used for index construction. The purpose is to allow a caller to place the vectors into native memory directly, instead of requiring the caller to load all the vectors into the heap (e.g. with a float[][]).

Public Members

ofArray

1
static CuVSMatrix ofArray(float[][] vectors)

Creates a dataset from an on-heap array of vectors. This method will allocate an additional MemorySegment to hold the graph data.

Source: java/cuvs-java/src/main/java/com/nvidia/cuvs/CuVSMatrix.java:46

ofArray

1
static CuVSMatrix ofArray(int[][] vectors)

Creates a dataset from an on-heap array of vectors. This method will allocate an additional MemorySegment to hold the graph data.

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

ofArray

1
static CuVSMatrix ofArray(byte[][] vectors)

Creates a dataset from an on-heap array of vectors. This method will allocate an additional MemorySegment to hold the graph data.

Source: java/cuvs-java/src/main/java/com/nvidia/cuvs/CuVSMatrix.java:66

addVector

1
void addVector(float[] vector)

Adds a single vector to the matrix.

Parameters

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

addVector

1
void addVector(byte[] vector)

Adds a single vector to the matrix.

Parameters

Source: java/cuvs-java/src/main/java/com/nvidia/cuvs/CuVSMatrix.java:87

addVector

1
void addVector(int[] vector)

Adds a single vector to the matrix.

Parameters

Source: java/cuvs-java/src/main/java/com/nvidia/cuvs/CuVSMatrix.java:94

hostBuilder

1
static Builder<CuVSHostMatrix> hostBuilder(long size, long columns, DataType dataType)

Returns a builder to create a new instance of a host-memory matrix

Parameters

Returns

a builder for creating a CuVSHostMatrix

Source: java/cuvs-java/src/main/java/com/nvidia/cuvs/CuVSMatrix.java:107

hostBuilder

1
static Builder<CuVSHostMatrix> hostBuilder( long size, long columns, int rowStride, int columnStride, DataType dataType)

Returns a builder to create a new instance of a host-memory matrix

Parameters

Returns

a builder for creating a CuVSDeviceMatrix

Source: java/cuvs-java/src/main/java/com/nvidia/cuvs/CuVSMatrix.java:121

deviceBuilder

1
static Builder<CuVSDeviceMatrix> deviceBuilder( CuVSResources resources, long size, long columns, DataType dataType)

Returns a builder to create a new instance of a dataset

Parameters

Returns

a builder for creating a CuVSDeviceMatrix

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

deviceBuilder

1
static Builder<CuVSDeviceMatrix> deviceBuilder( CuVSResources resources, long size, long columns, int rowStride, int columnStride, DataType dataType)

Returns a builder to create a new instance of a dataset

Parameters

Returns

a builder for creating a CuVSDeviceMatrix

Source: java/cuvs-java/src/main/java/com/nvidia/cuvs/CuVSMatrix.java:152

size

1
long size()

Gets the size of the dataset

Returns

Size of the dataset

Source: java/cuvs-java/src/main/java/com/nvidia/cuvs/CuVSMatrix.java:168

columns

1
long columns()

Gets the number of columns in the Dataset (e.g. the dimensions of the vectors in this dataset, or the graph degree for the graph represented as a list of neighbours

Returns

Dimensions of the vectors in the dataset

Source: java/cuvs-java/src/main/java/com/nvidia/cuvs/CuVSMatrix.java:176

dataType

1
DataType dataType()

Gets the element type

Returns

a DataType describing the matrix element type

Source: java/cuvs-java/src/main/java/com/nvidia/cuvs/CuVSMatrix.java:183

getRow

1
RowView getRow(long row)

Get a view (0-copy) of the row data, as a list of integers (32 bit)

Parameters

Source: java/cuvs-java/src/main/java/com/nvidia/cuvs/CuVSMatrix.java:190

toArray

1
void toArray(int[][] array)

Copies the content of this dataset to an on-heap Java matrix (array of arrays).

Parameters

Source: java/cuvs-java/src/main/java/com/nvidia/cuvs/CuVSMatrix.java:198

toArray

1
void toArray(float[][] array)

Copies the content of this dataset to an on-heap Java matrix (array of arrays).

Parameters

Source: java/cuvs-java/src/main/java/com/nvidia/cuvs/CuVSMatrix.java:206

toArray

1
void toArray(byte[][] array)

Copies the content of this dataset to an on-heap Java matrix (array of arrays).

Parameters

Source: java/cuvs-java/src/main/java/com/nvidia/cuvs/CuVSMatrix.java:214

toHost

1
void toHost(CuVSHostMatrix hostMatrix)

Fills the provided, pre-allocated host matrix with data from this matrix. The content of the provided host matrix will be overwritten; the 2 matrices must have the same element type and dimension.

Parameters

Source: java/cuvs-java/src/main/java/com/nvidia/cuvs/CuVSMatrix.java:223

toHost

1
CuVSHostMatrix toHost()

Returns a host matrix; if the matrix is already a host matrix, a “weak” reference to the same host memory is returned. If the matrix is a device matrix, a newly allocated matrix will be populated with data from the device matrix. The returned host matrix will need to be managed by the caller, which will be responsible to call CuVSMatrix#close() to free its resources when done.

Source: java/cuvs-java/src/main/java/com/nvidia/cuvs/CuVSMatrix.java:232

toDevice

1
void toDevice(CuVSDeviceMatrix deviceMatrix, CuVSResources cuVSResources)

Fills the provided, pre-allocated device matrix with data from this matrix. The content of the provided device matrix will be overwritten; the 2 matrices must have the same element type and dimension.

Parameters

Source: java/cuvs-java/src/main/java/com/nvidia/cuvs/CuVSMatrix.java:241

toDevice

1
CuVSDeviceMatrix toDevice(CuVSResources cuVSResources)

Returns a device matrix; if this matrix is already a device matrix, a “weak” reference to the same host memory is returned. If the matrix is a host matrix, a newly allocated matrix will be populated with data from the host matrix. The returned device matrix will need to be managed by the caller, which will be responsible to call CuVSMatrix#close() to free its resources when done.

Source: java/cuvs-java/src/main/java/com/nvidia/cuvs/CuVSMatrix.java:250

Source: java/cuvs-java/src/main/java/com/nvidia/cuvs/CuVSMatrix.java:17