From 9c79e3d5b1e0bd81c37e6a006a5b2340139d41b1 Mon Sep 17 00:00:00 2001 From: divyegala <divyegala@gmail.com> Date: Wed, 20 Mar 2024 14:28:42 -0700 Subject: [PATCH] RAFT 24.04 API changes (#3282) Summary: Pull Request resolved: https://github.com/facebookresearch/faiss/pull/3282 Reviewed By: junjieqi Differential Revision: D55153617 Pulled By: algoriddle fbshipit-source-id: 7b1ab24a6b0fbe002a0d8358078d014b1556044a --- faiss/gpu/impl/RaftIVFFlat.cu | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/faiss/gpu/impl/RaftIVFFlat.cu b/faiss/gpu/impl/RaftIVFFlat.cu index 1e310723d0..0906a60f46 100644 --- a/faiss/gpu/impl/RaftIVFFlat.cu +++ b/faiss/gpu/impl/RaftIVFFlat.cu @@ -403,7 +403,8 @@ void RaftIVFFlat::copyInvertedListsFrom(const InvertedLists* ivf) { } // Update the pointers and the sizes - raft_knn_index.value().recompute_internal_state(raft_handle); + raft::neighbors::ivf_flat::helpers::recompute_internal_state( + raft_handle, &(raft_knn_index.value())); for (size_t i = 0; i < nlist; ++i) { size_t listSize = ivf->list_size(i);