From f14be712214b975a6e99896611677e184fc2454d Mon Sep 17 00:00:00 2001
From: rhdong <hrong@nvidia.com>
Date: Thu, 3 Oct 2024 16:50:16 -0700
Subject: [PATCH] revert: update_dataset on strided matrix

---
 cpp/include/cuvs/neighbors/cagra.hpp            | 7 +++++--
 cpp/src/neighbors/detail/cagra/cagra_search.cuh | 1 +
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/cpp/include/cuvs/neighbors/cagra.hpp b/cpp/include/cuvs/neighbors/cagra.hpp
index 5b7a5ab0f..83d9eec12 100644
--- a/cpp/include/cuvs/neighbors/cagra.hpp
+++ b/cpp/include/cuvs/neighbors/cagra.hpp
@@ -432,8 +432,11 @@ struct index : cuvs::neighbors::index {
   void update_dataset(raft::resources const& res,
                       raft::device_matrix_view<const T, int64_t, raft::layout_stride> dataset)
   {
-    contiguous_dataset_ =
-      raft::make_device_matrix_view(dataset.data_handle(), dataset.extent(0), dataset.extent(1));
+    contiguous_dataset_ = std::monostate{};
+    if (dataset.stride(0) == dataset.extent(1) && dataset.stride(1) == 1) {
+      contiguous_dataset_ =
+        raft::make_device_matrix_view(dataset.data_handle(), dataset.extent(0), dataset.extent(1));
+    }
     dataset_ = make_aligned_dataset(res, dataset, 16);
   }
 
diff --git a/cpp/src/neighbors/detail/cagra/cagra_search.cuh b/cpp/src/neighbors/detail/cagra/cagra_search.cuh
index 5a1b764d0..ba0d82831 100644
--- a/cpp/src/neighbors/detail/cagra/cagra_search.cuh
+++ b/cpp/src/neighbors/detail/cagra/cagra_search.cuh
@@ -177,6 +177,7 @@ void search_main(raft::resources const& res,
         std::get_if<raft::device_matrix_view<const T, int64_t, raft::row_major>>(&dataset_view);
 
       if (brute_force_dataset) {
+        RAFT_LOG_DEBUG("CAGRA is switching to brute force with sparsity:%d", sparsity);
         auto brute_force_idx =
           cuvs::neighbors::brute_force::build(res, *brute_force_dataset, index.metric());
         cuvs::neighbors::brute_force::search(