RAFT 24.04 API changes (#3282)

Summary: Pull Request resolved: #3282 Reviewed By: junjieqi Differential Revision: D55153617 Pulled By: algoriddle fbshipit-source-id: 7b1ab24a6b0fbe002a0d8358078d014b1556044a
facebookresearch · Mar 20, 2024 · 9c79e3d · 9c79e3d
1 parent 5483f21
commit 9c79e3d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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);