Add missing Thrust includes

cpp/CMakeLists.txt

@@ -30,6 +30,9 @@ rapids_cuda_init_architectures(RAFT)
 
 project(RAFT VERSION 22.08.00 LANGUAGES CXX CUDA)
 
+include(${rapids-cmake-dir}/cpm/package_override.cmake)
+rapids_cpm_package_override(${CMAKE_CURRENT_SOURCE_DIR}/versions.json)
+
 # Needed because GoogleBenchmark changes the state of FindThreads.cmake, causing subsequent runs to
 # have different values for the `Threads::Threads` target. Setting this flag ensures
 # `Threads::Threads` is the same value in first run and subsequent runs.

cpp/include/raft/cluster/detail/kmeans.cuh

@@ -22,12 +22,19 @@
 
 #include <cuda.h>
 #include <thrust/binary_search.h>
+#include <thrust/device_ptr.h>
+#include <thrust/fill.h>
+#include <thrust/find.h>
+#include <thrust/functional.h>
 #include <thrust/gather.h>
 #include <thrust/iterator/constant_iterator.h>
+#include <thrust/memory.h>
 #include <thrust/random.h>
 #include <thrust/reduce.h>
+#include <thrust/scan.h>
 #include <thrust/sequence.h>
 #include <thrust/sort.h>
+#include <thrust/transform.h>
 
 #include <raft/cuda_utils.cuh>
 #include <raft/cudart_utils.h>

cpp/include/raft/core/span.hpp

@@ -19,6 +19,7 @@
 #include <cinttypes>  // size_t
 #include <cstddef>    // std::byte
 #include <raft/detail/span.hpp>
+#include <thrust/distance.h>
 #include <thrust/functional.h>
 #include <thrust/host_vector.h>  // __host__ __device__
 #include <thrust/iterator/reverse_iterator.h>

cpp/include/raft/lap/detail/lap_functions.cuh

@@ -32,6 +32,8 @@
 #include <rmm/device_scalar.hpp>
 #include <rmm/device_uvector.hpp>
 
+#include <thrust/execution_policy.h>
+#include <thrust/fill.h>
 #include <thrust/reduce.h>
 #include <thrust/scan.h>
 

cpp/include/raft/lap/detail/lap_kernels.cuh

@@ -29,6 +29,7 @@
 #include <raft/cudart_utils.h>
 #include <raft/handle.hpp>
 
+#include <thrust/execution_policy.h>
 #include <thrust/for_each.h>
 
 #include <cstddef>

cpp/include/raft/lap/lap.cuh

@@ -31,6 +31,9 @@
 #include <raft/handle.hpp>
 #include <rmm/device_uvector.hpp>
 
+#include <thrust/execution_policy.h>
+#include <thrust/fill.h>
+
 #include "detail/d_structs.h"
 #include "detail/lap_functions.cuh"
 

cpp/include/raft/lap/lap.hpp

@@ -36,6 +36,9 @@
 #include <raft/handle.hpp>
 #include <rmm/device_uvector.hpp>
 
+#include <thrust/execution_policy.h>
+#include <thrust/fill.h>
+
 #include "detail/d_structs.h"
 #include "detail/lap_functions.cuh"
 

cpp/include/raft/linalg/detail/add.cuh

@@ -22,6 +22,8 @@
 #include <raft/linalg/binary_op.cuh>
 #include <raft/linalg/unary_op.cuh>
 
+#include <thrust/functional.h>
+
 namespace raft {
 namespace linalg {
 namespace detail {

cpp/include/raft/linalg/detail/eltwise.cuh

@@ -21,6 +21,8 @@
 #include <raft/linalg/binary_op.cuh>
 #include <raft/linalg/unary_op.cuh>
 
+#include <thrust/functional.h>
+
 namespace raft {
 namespace linalg {
 namespace detail {

cpp/include/raft/matrix/detail/matrix.cuh

@@ -22,6 +22,7 @@
 #include <rmm/exec_policy.hpp>
 
 #include <thrust/for_each.h>
+#include <thrust/iterator/counting_iterator.h>
 
 #include <algorithm>
 #include <cstddef>

cpp/include/raft/sparse/distance/detail/coo_spmv_strategies/coo_mask_row_iterators.cuh

@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2021, NVIDIA CORPORATION.
+ * Copyright (c) 2021-2022, NVIDIA CORPORATION.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -21,6 +21,9 @@
 
 #include <rmm/device_uvector.hpp>
 
+#include <thrust/scan.h>
+#include <thrust/transform.h>
+
 namespace raft {
 namespace sparse {
 namespace distance {

cpp/include/raft/sparse/distance/detail/coo_spmv_strategies/hash_strategy.cuh

@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2021, NVIDIA CORPORATION.
+ * Copyright (c) 2021-2022, NVIDIA CORPORATION.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -20,6 +20,9 @@
 
 #include <cuco/static_map.cuh>
 
+#include <thrust/copy.h>
+#include <thrust/iterator/counting_iterator.h>
+
 // this is needed by cuco as key, value must be bitwise comparable.
 // compilers don't declare float/double as bitwise comparable
 // but that is too strict

cpp/include/raft/sparse/distance/detail/l2_distance.cuh

@@ -31,6 +31,9 @@
 
 #include <nvfunctional>
 
+#include <thrust/for_each.h>
+#include <thrust/iterator/counting_iterator.h>
+
 #include <algorithm>
 
 namespace raft {

cpp/include/raft/sparse/hierarchy/detail/agglomerative.cuh

@@ -24,7 +24,14 @@
 
 #include <thrust/device_ptr.h>
 #include <thrust/execution_policy.h>
+#include <thrust/extrema.h>
+#include <thrust/fill.h>
+#include <thrust/for_each.h>
+#include <thrust/functional.h>
+#include <thrust/iterator/counting_iterator.h>
+#include <thrust/iterator/zip_iterator.h>
 #include <thrust/sort.h>
+#include <thrust/tuple.h>
 
 #include <cstddef>
 

cpp/include/raft/sparse/hierarchy/detail/connectivities.cuh

@@ -29,6 +29,10 @@
 #include <raft/sparse/hierarchy/common.h>
 #include <raft/sparse/selection/knn_graph.cuh>
 
+#include <thrust/iterator/zip_iterator.h>
+#include <thrust/transform.h>
+#include <thrust/tuple.h>
+
 #include <limits>
 
 namespace raft {

cpp/include/raft/sparse/mst/detail/mst_solver_inl.cuh

@@ -26,12 +26,20 @@
 #include <rmm/device_uvector.hpp>
 
 #include <iostream>
+#include <thrust/copy.h>
 #include <thrust/device_ptr.h>
 #include <thrust/execution_policy.h>
+#include <thrust/fill.h>
+#include <thrust/functional.h>
 #include <thrust/host_vector.h>
+#include <thrust/iterator/zip_iterator.h>
 #include <thrust/reduce.h>
+#include <thrust/sequence.h>
 #include <thrust/sort.h>
 #include <thrust/transform.h>
+#include <thrust/transform_reduce.h>
+#include <thrust/tuple.h>
+#include <thrust/unique.h>
 
 namespace raft {
 namespace mst {

cpp/include/raft/sparse/op/detail/filter.cuh

@@ -25,6 +25,7 @@
 #include <rmm/exec_policy.hpp>
 
 #include <thrust/device_ptr.h>
+#include <thrust/reduce.h>
 #include <thrust/scan.h>
 
 #include <cuda_runtime.h>

cpp/include/raft/sparse/op/detail/sort.h

@@ -24,7 +24,10 @@
 #include <rmm/exec_policy.hpp>
 
 #include <thrust/device_ptr.h>
+#include <thrust/iterator/zip_iterator.h>
 #include <thrust/scan.h>
+#include <thrust/sort.h>
+#include <thrust/tuple.h>
 
 #include <cusparse_v2.h>
 

cpp/include/raft/sparse/selection/detail/connect_components.cuh

@@ -28,8 +28,14 @@
 
 #include <rmm/device_uvector.hpp>
 #include <rmm/exec_policy.hpp>
+#include <thrust/copy.h>
 #include <thrust/device_ptr.h>
+#include <thrust/iterator/counting_iterator.h>
+#include <thrust/iterator/zip_iterator.h>
+#include <thrust/scan.h>
 #include <thrust/sort.h>
+#include <thrust/transform.h>
+#include <thrust/tuple.h>
 
 #include <cub/cub.cuh>
 

cpp/include/raft/spatial/knn/detail/ball_cover.cuh

@@ -42,10 +42,15 @@
 #include <faiss/gpu/utils/Select.cuh>
 #include <faiss/utils/Heap.h>
 
+#include <thrust/fill.h>
+#include <thrust/for_each.h>
 #include <thrust/functional.h>
+#include <thrust/iterator/counting_iterator.h>
+#include <thrust/iterator/zip_iterator.h>
 #include <thrust/reduce.h>
 #include <thrust/sequence.h>
 #include <thrust/sort.h>
+#include <thrust/tuple.h>
 
 namespace raft {
 namespace spatial {

cpp/include/raft/spatial/knn/detail/ball_cover/common.cuh

@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2021, NVIDIA CORPORATION.
+ * Copyright (c) 2021-2022, NVIDIA CORPORATION.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -19,6 +19,7 @@
 #include "../haversine_distance.cuh"
 #include <cstdint>
 #include <thrust/functional.h>
+#include <thrust/tuple.h>
 
 namespace raft {
 namespace spatial {

cpp/include/raft/spatial/knn/detail/ball_cover/registers.cuh

@@ -32,6 +32,8 @@
 #include <faiss/gpu/utils/Select.cuh>
 #include <faiss/utils/Heap.h>
 
+#include <thrust/fill.h>
+
 namespace raft {
 namespace spatial {
 namespace knn {

cpp/include/raft/spectral/detail/matrix_wrappers.hpp

@@ -21,8 +21,10 @@
 #include <raft/sparse/detail/cusparse_wrappers.h>
 #include <rmm/device_uvector.hpp>
 
+#include <thrust/execution_policy.h>
 #include <thrust/fill.h>
 #include <thrust/reduce.h>
+#include <thrust/system/cuda/execution_policy.h>
 
 #include <algorithm>
 

cpp/include/raft/spectral/detail/spectral_util.cuh

@@ -21,9 +21,15 @@
 #include <raft/linalg/detail/cublas_wrappers.hpp>
 #include <raft/spectral/matrix_wrappers.hpp>
 
+#include <thrust/device_ptr.h>
 #include <thrust/fill.h>
+#include <thrust/for_each.h>
+#include <thrust/functional.h>
+#include <thrust/iterator/constant_iterator.h>
+#include <thrust/iterator/zip_iterator.h>
 #include <thrust/reduce.h>
 #include <thrust/transform.h>
+#include <thrust/tuple.h>
 
 #include <algorithm>
 

cpp/include/raft/stats/detail/adjusted_rand_index.cuh

@@ -33,6 +33,10 @@
 #include <rmm/device_scalar.hpp>
 #include <rmm/device_uvector.hpp>
 
+#include <thrust/device_ptr.h>
+#include <thrust/execution_policy.h>
+#include <thrust/extrema.h>
+
 namespace raft {
 namespace stats {
 namespace detail {

cpp/include/raft/stats/detail/batched/silhouette_score.cuh

@@ -22,6 +22,8 @@
 #include <rmm/device_uvector.hpp>
 #include <rmm/exec_policy.hpp>
 #include <thrust/device_vector.h>
+#include <thrust/fill.h>
+#include <thrust/reduce.h>
 
 namespace raft {
 namespace stats {

cpp/include/raft/stats/detail/contingencyMatrix.cuh

@@ -20,6 +20,8 @@
 #include <raft/cudart_utils.h>
 
 #include <thrust/device_ptr.h>
+#include <thrust/execution_policy.h>
+#include <thrust/extrema.h>
 #include <thrust/reduce.h>
 
 #include <cub/cub.cuh>

cpp/include/raft/stats/detail/scores.cuh

@@ -26,7 +26,9 @@
 #include <raft/stats/mean.cuh>
 #include <rmm/device_scalar.hpp>
 #include <rmm/device_uvector.hpp>
+#include <thrust/count.h>
 #include <thrust/device_ptr.h>
+#include <thrust/execution_policy.h>
 #include <thrust/reduce.h>
 
 #define N_THREADS 512

cpp/test/linalg/reduce.cuh

@@ -23,6 +23,7 @@
 #include <rmm/device_uvector.hpp>
 
 #include <thrust/device_ptr.h>
+#include <thrust/memory.h>
 
 namespace raft {
 namespace linalg {

cpp/test/matrix/matrix.cu

@@ -21,6 +21,10 @@
 #include <raft/random/rng.cuh>
 #include <rmm/device_uvector.hpp>
 
+#include <thrust/copy.h>
+#include <thrust/device_ptr.h>
+#include <thrust/iterator/counting_iterator.h>
+
 namespace raft {
 namespace matrix {
 

cpp/test/mdarray.cu

@@ -23,8 +23,10 @@
 #include <rmm/device_vector.hpp>
 #include <rmm/exec_policy.hpp>
 #include <thrust/device_vector.h>
+#include <thrust/execution_policy.h>
 #include <thrust/for_each.h>
 #include <thrust/iterator/counting_iterator.h>
+#include <thrust/sequence.h>
 
 namespace {
 namespace stdex = std::experimental;

cpp/test/mst.cu

@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2020-2021, NVIDIA CORPORATION.
+ * Copyright (c) 2020-2022, NVIDIA CORPORATION.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -32,6 +32,10 @@
 
 #include <bits/stdc++.h>
 
+#include <thrust/execution_policy.h>
+#include <thrust/memory.h>
+#include <thrust/reduce.h>
+
 #include <cstddef>
 #include <vector>
 

cpp/test/random/make_regression.cu

@@ -16,6 +16,7 @@
 
 #include <gtest/gtest.h>
 #include <thrust/count.h>
+#include <thrust/device_ptr.h>
 #include <thrust/device_vector.h>
 
 #include "../test_utils.h"