diff --git a/cpp/src/hdbscan/detail/reachability_faiss.cuh b/cpp/src/hdbscan/detail/reachability_faiss.cuh index a4131a3308..804c8b500d 100644 --- a/cpp/src/hdbscan/detail/reachability_faiss.cuh +++ b/cpp/src/hdbscan/detail/reachability_faiss.cuh @@ -24,7 +24,6 @@ #pragma once #include -#include #include #include @@ -41,6 +40,7 @@ #include #include +#include #include @@ -212,7 +212,7 @@ void mutual_reachability_knn_l2(const raft::handle_t& handle, auto stream = handle.get_stream(); faiss::gpu::DeviceScope ds(device); - faiss::gpu::StandardGpuResources res; + raft::spatial::knn::RmmGpuResources res; res.noTempMemory(); res.setDefaultStream(device, stream); diff --git a/cpp/src_prims/selection/knn.cuh b/cpp/src_prims/selection/knn.cuh index bdba95d082..03fb6d83c3 100644 --- a/cpp/src_prims/selection/knn.cuh +++ b/cpp/src_prims/selection/knn.cuh @@ -35,7 +35,6 @@ #include #include #include -#include #include #include #include