From cf0db9171f8b1c5b1fe07e8046ac3897a49e0f7d Mon Sep 17 00:00:00 2001 From: tsuki <12711693+enp1s0@users.noreply.github.com> Date: Thu, 14 Mar 2024 00:49:15 +0900 Subject: [PATCH] Update cpp/include/raft/neighbors/detail/cagra/cagra_search.cuh Co-authored-by: Artem M. Chirkin <9253178+achirkin@users.noreply.github.com> --- cpp/include/raft/neighbors/detail/cagra/cagra_search.cuh | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/cpp/include/raft/neighbors/detail/cagra/cagra_search.cuh b/cpp/include/raft/neighbors/detail/cagra/cagra_search.cuh index c65ac502ce..0b9212d554 100644 --- a/cpp/include/raft/neighbors/detail/cagra/cagra_search.cuh +++ b/cpp/include/raft/neighbors/detail/cagra/cagra_search.cuh @@ -266,10 +266,10 @@ void search_main(raft::resources const& res, // determined here. They are set just before kernel launch. using dataset_desc_t = standard_dataset_descriptor_t; // Search using a plain (strided) row-major dataset - const dataset_desc_t dataset_desc(internal_dataset.data_handle(), - internal_dataset.extent(0), - internal_dataset.extent(1), - internal_dataset.stride(0)); +const dataset_desc_t dataset_desc(strided_dset->view().data_handle(), + strided_dset->n_rows(), + strided_dset->dim(), + strided_dset->stride()); search_main_core( res, params, dataset_desc, graph_internal, queries, neighbors, distances, sample_filter);