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);