From b237bee4e561eece8dc00755831f9f9f2c400ea4 Mon Sep 17 00:00:00 2001 From: Igor Motov Date: Thu, 30 Jul 2026 19:02:35 -1000 Subject: [PATCH 1/2] Fix cuvs-lucene to work with NVIDIA/cuvs#1846 --- RAPIDS_BRANCH | 2 +- .../cuvs/lucene/CuVS2510GPUVectorsWriter.java | 19 +++++++++++-------- 2 files changed, 12 insertions(+), 9 deletions(-) diff --git a/RAPIDS_BRANCH b/RAPIDS_BRANCH index 8ce6b738..e56d0310 100644 --- a/RAPIDS_BRANCH +++ b/RAPIDS_BRANCH @@ -1 +1 @@ -release/26.08 +pull-request/1846 diff --git a/src/main/java/com/nvidia/cuvs/lucene/CuVS2510GPUVectorsWriter.java b/src/main/java/com/nvidia/cuvs/lucene/CuVS2510GPUVectorsWriter.java index e8fb304e..97fedf92 100644 --- a/src/main/java/com/nvidia/cuvs/lucene/CuVS2510GPUVectorsWriter.java +++ b/src/main/java/com/nvidia/cuvs/lucene/CuVS2510GPUVectorsWriter.java @@ -1,5 +1,5 @@ /* - * SPDX-FileCopyrightText: Copyright (c) 2025-2026, NVIDIA CORPORATION. + * SPDX-FileCopyrightText: Copyright (c) 2025-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved. * SPDX-License-Identifier: Apache-2.0 */ package com.nvidia.cuvs.lucene; @@ -243,13 +243,16 @@ private void writeFieldInternal(FieldInfo fieldInfo, List vectors) thro private void writeCagraIndex(OutputStream os, CuVSMatrix dataset) throws Throwable { CagraIndexParams params = CagraIndexParamsFactory.create(gpuSearchParams, dataset.size(), dataset.columns()); - CagraIndex index = - CagraIndex.newBuilder(getCuVSResourcesInstance()) - .withDataset(dataset) - .withIndexParams(params) - .build(); - index.serialize(os); - index.close(); + try (CagraIndex index = + CagraIndex.newBuilder(getCuVSResourcesInstance()) + .withDataset(dataset) + .withIndexParams(params) + .build(); + var deviceVectors = dataset.toDevice(getCuVSResourcesInstance()); + var indexDataset = index.makePaddedDataset(deviceVectors)) { + index.updateDataset(indexDataset); + index.serialize(os); + } } /** From bcf618ef327007e29c58115d3d389d1bf63592c1 Mon Sep 17 00:00:00 2001 From: Igor Motov Date: Thu, 30 Jul 2026 19:29:25 -1000 Subject: [PATCH 2/2] Switch RAPIDS_BRANCH to release/26.08 --- RAPIDS_BRANCH | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/RAPIDS_BRANCH b/RAPIDS_BRANCH index e56d0310..8ce6b738 100644 --- a/RAPIDS_BRANCH +++ b/RAPIDS_BRANCH @@ -1 +1 @@ -pull-request/1846 +release/26.08