From ab0e03be112aebc830873f1c98f7739fd0afd660 Mon Sep 17 00:00:00 2001
From: Allard Hendriksen <allard@allardhendriksen.nl>
Date: Tue, 18 Apr 2023 17:08:09 +0200
Subject: [PATCH] Add missing includes of rmm/mr/device/per_device_resource.hpp
 (#5369)

Some files use `rmm::mr::get_current_device_resource()` but not do not include `rmm/mr/device/per_device_resource.hpp` . This PR fixes that.

The missing include was surfaced by https://github.com/rapidsai/cuml/pull/5363.

This PR is necessary to prevent breakage when https://github.com/rapidsai/raft/pull/1415 is merged.

Authors:
  - Allard Hendriksen (https://github.com/ahendriksen)

Approvers:
  - Corey J. Nolet (https://github.com/cjnolet)

URL: https://github.com/rapidsai/cuml/pull/5369
---
 cpp/bench/common/ml_benchmark.hpp      | 1 +
 cpp/examples/symreg/symreg_example.cpp | 3 ++-
 cpp/include/cuml/tsa/arima_common.h    | 1 +
 cpp/src/genetic/genetic.cu             | 1 +
 cpp/src/svm/linear.cu                  | 1 +
 cpp/test/mg/knn.cu                     | 2 ++
 cpp/test/sg/genetic/evolution_test.cu  | 1 +
 cpp/test/sg/genetic/program_test.cu    | 1 +
 8 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/cpp/bench/common/ml_benchmark.hpp b/cpp/bench/common/ml_benchmark.hpp
index fb90d64aca..17051e1d05 100644
--- a/cpp/bench/common/ml_benchmark.hpp
+++ b/cpp/bench/common/ml_benchmark.hpp
@@ -22,6 +22,7 @@
 #include <cuml/common/utils.hpp>
 #include <memory>
 #include <raft/util/cudart_utils.hpp>
+#include <rmm/mr/device/per_device_resource.hpp>
 #include <sstream>
 #include <string>
 #include <vector>
diff --git a/cpp/examples/symreg/symreg_example.cpp b/cpp/examples/symreg/symreg_example.cpp
index 7e8359efe9..9bca626e97 100644
--- a/cpp/examples/symreg/symreg_example.cpp
+++ b/cpp/examples/symreg/symreg_example.cpp
@@ -31,6 +31,7 @@
 #include <raft/util/cudart_utils.hpp>
 #include <rmm/device_scalar.hpp>
 #include <rmm/device_uvector.hpp>
+#include <rmm/mr/device/per_device_resource.hpp>
 
 // Namespace alias
 namespace cg = cuml::genetic;
@@ -340,4 +341,4 @@ int main(int argc, char* argv[])
   CUDA_RT_CALL(cudaEventDestroy(start));
   CUDA_RT_CALL(cudaEventDestroy(stop));
   return 0;
-}
\ No newline at end of file
+}
diff --git a/cpp/include/cuml/tsa/arima_common.h b/cpp/include/cuml/tsa/arima_common.h
index 597d05aa57..4e34358ae7 100644
--- a/cpp/include/cuml/tsa/arima_common.h
+++ b/cpp/include/cuml/tsa/arima_common.h
@@ -21,6 +21,7 @@
 #include <algorithm>
 
 #include <raft/util/cudart_utils.hpp>
+#include <rmm/mr/device/per_device_resource.hpp>
 #include <thrust/execution_policy.h>
 #include <thrust/for_each.h>
 #include <thrust/iterator/counting_iterator.h>
diff --git a/cpp/src/genetic/genetic.cu b/cpp/src/genetic/genetic.cu
index 3cc6680ce5..0db6954799 100644
--- a/cpp/src/genetic/genetic.cu
+++ b/cpp/src/genetic/genetic.cu
@@ -34,6 +34,7 @@
 
 #include <device_launch_parameters.h>
 #include <rmm/device_uvector.hpp>
+#include <rmm/mr/device/per_device_resource.hpp>
 
 namespace cuml {
 namespace genetic {
diff --git a/cpp/src/svm/linear.cu b/cpp/src/svm/linear.cu
index 20c47078b9..bc472fa1a5 100644
--- a/cpp/src/svm/linear.cu
+++ b/cpp/src/svm/linear.cu
@@ -40,6 +40,7 @@
 #include <raft/matrix/matrix.cuh>
 #include <raft/util/cuda_utils.cuh>
 #include <rmm/device_uvector.hpp>
+#include <rmm/mr/device/per_device_resource.hpp>
 #include <thrust/copy.h>
 #include <thrust/device_ptr.h>
 #include <thrust/execution_policy.h>
diff --git a/cpp/test/mg/knn.cu b/cpp/test/mg/knn.cu
index dd6f903758..d7b4d5c8de 100644
--- a/cpp/test/mg/knn.cu
+++ b/cpp/test/mg/knn.cu
@@ -25,6 +25,8 @@
 
 #include <raft/util/cuda_utils.cuh>
 
+#include <rmm/mr/device/per_device_resource.hpp>
+
 namespace ML {
 namespace KNN {
 namespace opg {
diff --git a/cpp/test/sg/genetic/evolution_test.cu b/cpp/test/sg/genetic/evolution_test.cu
index 41f1122c58..ff3c746fe0 100644
--- a/cpp/test/sg/genetic/evolution_test.cu
+++ b/cpp/test/sg/genetic/evolution_test.cu
@@ -26,6 +26,7 @@
 #include <raft/core/handle.hpp>
 #include <raft/util/cudart_utils.hpp>
 #include <rmm/device_uvector.hpp>
+#include <rmm/mr/device/per_device_resource.hpp>
 #include <test_utils.h>
 #include <vector>
 
diff --git a/cpp/test/sg/genetic/program_test.cu b/cpp/test/sg/genetic/program_test.cu
index 9296d7be0a..f072f107e4 100644
--- a/cpp/test/sg/genetic/program_test.cu
+++ b/cpp/test/sg/genetic/program_test.cu
@@ -24,6 +24,7 @@
 #include <raft/core/handle.hpp>
 #include <raft/util/cudart_utils.hpp>
 #include <rmm/device_uvector.hpp>
+#include <rmm/mr/device/per_device_resource.hpp>
 #include <test_utils.h>
 #include <vector>