RAFT 24.04 API changes (facebookresearch#3282)

divyegala · abhinavdangeti · commit ff1d8e103fab · 2024-07-11T19:50:07.000-06:00
Summary: Pull Request resolved: facebookresearch#3282 Reviewed By: junjieqi Differential Revision: D55153617 Pulled By: algoriddle fbshipit-source-id: 7b1ab24a6b0fbe002a0d8358078d014b1556044a
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);

Original file line number	Diff line number	Diff line change
`@@ -403,7 +403,8 @@ void RaftIVFFlat::copyInvertedListsFrom(const InvertedLists* ivf) {`
`403`	`403`	`}`
`404`	`404`
`405`	`405`	`// Update the pointers and the sizes`
`406`		`- raft_knn_index.value().recompute_internal_state(raft_handle);`
	`406`	`+ raft::neighbors::ivf_flat::helpers::recompute_internal_state(`
	`407`	`+ raft_handle, &(raft_knn_index.value()));`
`407`	`408`
`408`	`409`	`for (size_t i = 0; i < nlist; ++i) {`
`409`	`410`	`size_t listSize = ivf->list_size(i);`