Skip to content

Conversation

mayya-sharipova
Copy link
Contributor

@mayya-sharipova mayya-sharipova commented Oct 6, 2025

Update Nvidia CUVS java library to a newer version that includes strides

Co-authored-by: Brian Seeders [email protected]

@mayya-sharipova mayya-sharipova requested a review from a team as a code owner October 6, 2025 20:28
@mayya-sharipova mayya-sharipova added auto-backport Automatically create backport pull requests when merged :Search Relevance/Vectors Vector search v9.2.1 v9.3.0 labels Oct 6, 2025
@mayya-sharipova mayya-sharipova requested review from ldematte and removed request for a team October 6, 2025 20:28
@elasticsearchmachine
Copy link
Collaborator

Pinging @elastic/es-search-relevance (Team:Search Relevance)

@elasticsearchmachine elasticsearchmachine added the Team:Search Relevance Meta label for the Search Relevance team in Elasticsearch label Oct 6, 2025
Copy link
Contributor

@ldematte ldematte left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

A small change needed but looks good

Copy link
Contributor

@ldematte ldematte left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

CuVSMatrix.DataType dataType
) {
try {
return (CuVSMatrix) createDatasetWithStrides$mh.invokeExact(memorySegment, size, dimensions, rowStride, columnStride, dataType);
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

As we don't need columnStride and it's not really supported (yet) in cuvs-java, I think you might want to pass -1 here like you did in 14088bd. So you can remove all the int columnStride params from all these functions and just have rowStride.
But up to you (I'm fine either way).

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks, may be we can keep this here similar to cuvs DatasetHelper. and adjust it later if needed.

@mayya-sharipova mayya-sharipova merged commit 0b2682b into main Oct 7, 2025
35 checks passed
@mayya-sharipova mayya-sharipova deleted the cuvs-snapshot-update branch October 7, 2025 23:18
mayya-sharipova added a commit to mayya-sharipova/elasticsearch that referenced this pull request Oct 7, 2025
This updates GPU Plugin to a newer CUVS version

Co-authored-by: Brian Seeders <[email protected]>
@elasticsearchmachine
Copy link
Collaborator

💚 Backport successful

Status Branch Result
9.2

elasticsearchmachine pushed a commit that referenced this pull request Oct 8, 2025
This updates GPU Plugin to a newer CUVS version

Co-authored-by: Brian Seeders <[email protected]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
auto-backport Automatically create backport pull requests when merged >non-issue :Search Relevance/Vectors Vector search Team:Search Relevance Meta label for the Search Relevance team in Elasticsearch v9.2.1 v9.3.0
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants