a little too aggressive in deleting template instantiations...

rapidsai · Oct 16, 2024 · d1e01a2 · d1e01a2
1 parent ba6e5dc
commit d1e01a2
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/cpp/src/detail/collect_local_vertex_values_mg_v32_e32.cu b/cpp/src/detail/collect_local_vertex_values_mg_v32_e32.cu
@@ -15,7 +15,6 @@
  */
 
 #include "detail/collect_local_vertex_values.cuh"
-#include "detail/graph_partition_utils.cuh"
 
 #include <cugraph/graph_functions.hpp>
 #include <cugraph/utilities/shuffle_comm.cuh>

diff --git a/cpp/src/detail/collect_local_vertex_values_sg_v32_e32.cu b/cpp/src/detail/collect_local_vertex_values_sg_v32_e32.cu
@@ -15,7 +15,6 @@
  */
 
 #include "detail/collect_local_vertex_values.cuh"
-#include "detail/graph_partition_utils.cuh"
 
 #include <cugraph/graph_functions.hpp>
 #include <cugraph/utilities/shuffle_comm.cuh>

diff --git a/cpp/tests/utilities/thrust_wrapper.cu b/cpp/tests/utilities/thrust_wrapper.cu
@@ -169,6 +169,11 @@ sort_by_key<int32_t, int32_t>(raft::handle_t const& handle,
                               rmm::device_uvector<int32_t> const& keys,
                               rmm::device_uvector<int32_t> const& values);
 
+template std::tuple<rmm::device_uvector<int32_t>, rmm::device_uvector<int64_t>>
+sort_by_key<int32_t, int64_t>(raft::handle_t const& handle,
+                              rmm::device_uvector<int32_t> const& keys,
+                              rmm::device_uvector<int64_t> const& values);
+
 template std::tuple<rmm::device_uvector<int64_t>, rmm::device_uvector<float>>
 sort_by_key<int64_t, float>(raft::handle_t const& handle,
                             rmm::device_uvector<int64_t> const& keys,
@@ -255,6 +260,11 @@ sort_by_key<int32_t, int32_t>(raft::handle_t const& handle,
                               rmm::device_uvector<int32_t>&& keys,
                               rmm::device_uvector<int32_t>&& values);
 
+template std::tuple<rmm::device_uvector<int32_t>, rmm::device_uvector<int64_t>>
+sort_by_key<int32_t, int64_t>(raft::handle_t const& handle,
+                              rmm::device_uvector<int32_t>&& keys,
+                              rmm::device_uvector<int64_t>&& values);
+
 template std::tuple<rmm::device_uvector<int64_t>, rmm::device_uvector<float>>
 sort_by_key<int64_t, float>(raft::handle_t const& handle,
                             rmm::device_uvector<int64_t>&& keys,