Remove topk.cu, fix comments

rapidsai · Apr 18, 2023 · 99d08e5 · 99d08e5
1 parent cb5b7d0
commit 99d08e5
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 10 deletions.
diff --git a/cpp/CMakeLists.txt b/cpp/CMakeLists.txt
@@ -314,7 +314,6 @@ if(RAFT_COMPILE_LIBRARY)
     # src/neighbors/cagra/search_uint8_t_dim256_t8.cu
     # src/neighbors/cagra/search_uint8_t_dim512_t16.cu
     # src/neighbors/cagra/search_uint8_t_dim512_t32.cu
-    src/neighbors/cagra/topk.cu
     src/neighbors/refine_d_int64_t_float.cu
     src/neighbors/refine_d_int64_t_int8_t.cu
     src/neighbors/refine_d_int64_t_uint8_t.cu

diff --git a/cpp/include/raft/neighbors/specializations/cagra.cuh b/cpp/include/raft/neighbors/specializations/cagra.cuh
@@ -74,17 +74,18 @@ RAFT_INST(int8_t, uint32_t, memory_type::device, memory_type::host);
 
 #undef RAFT_INST
 
-// #define RAFT_INST(T, IdxT)                                      \
-//   extern template void search<T, IdxT>(                         \
-//     raft::device_resources const& handle,                       \
-//     const search_params& params,                                \
-//     const index<T, IdxT>& idx,                                  \
-//     raft::device_matrix_view<const T, IdxT, row_major> queries, \
-//     raft::device_matrix_view<IdxT, IdxT, row_major> neighbors,  \
-//     raft::device_matrix_view<float, IdxT, row_major> distances);
+/* #define RAFT_INST(T, IdxT)                                      \
+  extern template void search<T, IdxT>(                         \
+    raft::device_resources const& handle,                       \
+    const search_params& params,                                \
+    const index<T, IdxT>& idx,                                  \
+    raft::device_matrix_view<const T, IdxT, row_major> queries, \
+    raft::device_matrix_view<IdxT, IdxT, row_major> neighbors,  \
+    raft::device_matrix_view<float, IdxT, row_major> distances);
 
-// RAFT_INST(float, uint32_t)
+RAFT_INST(float, uint32_t)
 #undef RAFT_INST
+*/
 }  // namespace raft::neighbors::experimental::cagra
 namespace raft::neighbors::experimental::cagra::detail::single_cta_search {
 // extern template struct search<4, 128, float, uint32_t, float>;