diff --git a/cpp/include/raft/neighbors/detail/cagra/cagra_search.cuh b/cpp/include/raft/neighbors/detail/cagra/cagra_search.cuh index 0e20c56e53..836710bc04 100644 --- a/cpp/include/raft/neighbors/detail/cagra/cagra_search.cuh +++ b/cpp/include/raft/neighbors/detail/cagra/cagra_search.cuh @@ -253,8 +253,6 @@ void search_main(raft::resources const& res, auto graph_internal = raft::make_device_matrix_view( reinterpret_cast(graph.data_handle()), graph.extent(0), graph.extent(1)); - bool use_vpq = false; - // n_rows has the same type as the dataset index (the array extents type) using ds_idx_type = decltype(index.dataset().n_rows()); // Dispatch search parameters based on the dataset kind. @@ -269,7 +267,7 @@ 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(strided_dset->view().data_handle(), + const dataset_desc_t dataset_desc(strided_dset->view().data_handle(), strided_dset->n_rows(), strided_dset->dim(), strided_dset->stride()); @@ -279,12 +277,10 @@ const dataset_desc_t dataset_desc(strided_dset->view().data_handle(), } else if (auto* vpq_dset = dynamic_cast*>(&index.dataset()); vpq_dset != nullptr) { - use_vpq = true; // Search using a compressed dataset RAFT_FAIL("FP32 VPQ dataset support is coming soon"); } else if (auto* vpq_dset = dynamic_cast*>(&index.dataset()); vpq_dset != nullptr) { - use_vpq = true; lauch_vpq_search_main_core( res, vpq_dset, params, graph_internal, queries, neighbors, distances, sample_filter); } else if (auto* empty_dset = dynamic_cast*>(&index.dataset());