From eadcbf2892f466550922e16e3da237d853f5ff2e Mon Sep 17 00:00:00 2001 From: tsuki <12711693+enp1s0@users.noreply.github.com> Date: Mon, 26 Feb 2024 12:56:36 +0900 Subject: [PATCH] Update cpp/include/raft/neighbors/detail/cagra/add_nodes.cuh Co-authored-by: Tamas Bela Feher --- cpp/include/raft/neighbors/detail/cagra/add_nodes.cuh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/cpp/include/raft/neighbors/detail/cagra/add_nodes.cuh b/cpp/include/raft/neighbors/detail/cagra/add_nodes.cuh index 297cb9e4cc..f01919d11c 100644 --- a/cpp/include/raft/neighbors/detail/cagra/add_nodes.cuh +++ b/cpp/include/raft/neighbors/detail/cagra/add_nodes.cuh @@ -128,8 +128,8 @@ void add_node_core(raft::resources const& handle, // Step 1: Obtain K (=base_degree) nearest neighbors of the new vectors by CAGRA search // Create queries for (std::size_t i = 0; i < actual_batch_size; i++) { - raft::copy(queries.data_handle() + i * dim, - additional_dataset_view.data_handle() + (new_vec_id_offset + i) * dim, + raft::copy(&queries(i, 0), + &additional_dataset_view(new_vec_id_offset + i, dim), dim, raft::resource::get_cuda_stream(handle)); }