(Fix an issue of wrong variable name)

rapidsai · Mar 25, 2024 · 4e787d4 · 4e787d4
1 parent 7cd6f0e
commit 4e787d4
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/cpp/include/raft/neighbors/detail/cagra/cagra_search.cuh b/cpp/include/raft/neighbors/detail/cagra/cagra_search.cuh
@@ -90,12 +90,12 @@ void search_main_core(
   CagraSampleFilterT sample_filter = CagraSampleFilterT())
 {
   RAFT_LOG_DEBUG("# dataset size = %lu, dim = %lu\n",
-                 static_cast<size_t>(index.data().n_rows()),
-                 static_cast<size_t>(index.data().dim()));
+                 static_cast<size_t>(graph.data().n_rows()),
+                 static_cast<size_t>(graph.data().dim()));
   RAFT_LOG_DEBUG("# query size = %lu, dim = %lu\n",
                  static_cast<size_t>(queries.extent(0)),
                  static_cast<size_t>(queries.extent(1)));
-  RAFT_EXPECTS(queries.extent(1) == dataset_desc.dim, "Queries and index dim must match");
+  RAFT_EXPECTS(queries.extent(1) == dataset_desc.dim, "Queries and graph dim must match");
   const uint32_t topk = neighbors.extent(1);
 
   cudaDeviceProp deviceProp = resource::get_device_properties(res);