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>