More updates

rapidsai · Nov 15, 2024 · 4615f0b · 4615f0b
1 parent a5442fc
commit 4615f0b
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 6 deletions.
diff --git a/cpp/CMakeLists.txt b/cpp/CMakeLists.txt
@@ -373,6 +373,7 @@ if(BUILD_SHARED_LIBS)
     src/distance/distance.cu
     src/distance/kernel_gram.cu
     src/distance/pairwise_distance.cu
+    src/dimred/spectral.cu
     src/neighbors/brute_force.cu
     src/neighbors/cagra_build_float.cu
     src/neighbors/cagra_build_half.cu
@@ -462,7 +463,6 @@ if(BUILD_SHARED_LIBS)
     src/selection/select_k_float_int32_t.cu
     src/selection/select_k_float_uint32_t.cu
     src/selection/select_k_half_uint32_t.cu
-    src/sparse/embed/spectral.cu
     src/stats/silhouette_score.cu
     src/stats/trustworthiness_score.cu
     ${CUVS_MG_ALGOS}

diff --git a/cpp/include/cuvs/sparse/embed/spectral.hpp → cpp/include/cuvs/dimred/spectral.hpp b/cpp/include/cuvs/sparse/embed/spectral.hpp → cpp/include/cuvs/dimred/spectral.hpp
@@ -17,7 +17,7 @@
 #include <raft/core/device_coo_matrix.hpp>
 #include <raft/core/resources.hpp>
 
-namespace cuvs::sparse::cluster::spectral {
+namespace cuvs::dimred::spectral {
 
 /**
  * Given a COO formatted (symmetric) knn graph, this function computes the spectral embeddings
@@ -39,4 +39,4 @@ void fit(const raft::resources& handle,
          int n_components,
          raft::device_matrix_view<float, int> out,
          unsigned long long seed = 0L);
-};  // namespace cuvs::sparse::cluster::spectral
+};  // namespace cuvs::dimred::spectral
diff --git a/cpp/include/cuvs/distance/grammian.hpp b/cpp/include/cuvs/distance/grammian.hpp
@@ -16,6 +16,7 @@
 
 #pragma once
 
+#include <cublas_v2.h>
 #include <cuvs/distance/distance.hpp>
 #include <raft/core/device_csr_matrix.hpp>
 #include <raft/core/mdspan.hpp>

diff --git a/cpp/src/sparse/embed/spectral.cu → cpp/src/dimred/spectral.cu b/cpp/src/sparse/embed/spectral.cu → cpp/src/dimred/spectral.cu
@@ -14,12 +14,12 @@
  * limitations under the License.
  */
 
-#include "../cluster/detail/spectral.cuh"
+#include "../sparse/cluster/detail/spectral.cuh"
 #include <cuvs/sparse/embed/spectral.hpp>
 #include <raft/core/device_coo_matrix.hpp>
 #include <raft/core/resources.hpp>
 
-namespace cuvs::sparse::embed::spectral {
+namespace cuvs::dimred::spectral {
 
 /**
  * Given a COO formatted (symmetric) knn graph, this function computes the spectral embeddings
@@ -50,4 +50,4 @@ void fit(const raft::resources& handle,
     out.data_handle(),
     seed);
 }
-};  // namespace cuvs::sparse::embed::spectral
+};  // namespace cuvs::dimred::spectral