From 49ae78f42f272f54f234622da02ecec0da9112db Mon Sep 17 00:00:00 2001
From: Allison Piper <alliepiper16@gmail.com>
Date: Sat, 3 May 2025 10:08:29 -0400
Subject: [PATCH 1/2] Add missing header to host_memory.cpp

Discovered in CCCL CI: https://github.com/NVIDIA/cccl/actions/runs/14806960648/job/41576694188#step:6:2863
---
 cpp/src/utilities/host_memory.cpp | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/cpp/src/utilities/host_memory.cpp b/cpp/src/utilities/host_memory.cpp
index e41d772a479..1f11e1539f7 100644
--- a/cpp/src/utilities/host_memory.cpp
+++ b/cpp/src/utilities/host_memory.cpp
@@ -26,6 +26,8 @@
 #include <rmm/mr/device/pool_memory_resource.hpp>
 #include <rmm/mr/pinned_host_memory_resource.hpp>
 
+#include <atomic>
+
 namespace cudf {
 
 namespace {

From dfa0dbe686a0a5011feb7563512840538cdc9b69 Mon Sep 17 00:00:00 2001
From: Michael Schellenberger Costa <miscco@nvidia.com>
Date: Mon, 5 May 2025 08:06:12 +0200
Subject: [PATCH 2/2] Add missing includes

We dropped some includes from cub and this file started to break in our CI
---
 cpp/src/utilities/host_memory.cpp | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/cpp/src/utilities/host_memory.cpp b/cpp/src/utilities/host_memory.cpp
index e41d772a479..e6c2d95ec70 100644
--- a/cpp/src/utilities/host_memory.cpp
+++ b/cpp/src/utilities/host_memory.cpp
@@ -26,6 +26,13 @@
 #include <rmm/mr/device/pool_memory_resource.hpp>
 #include <rmm/mr/pinned_host_memory_resource.hpp>
 
+#include <algorithm>
+#include <atomic>
+#include <cstdint>
+#include <cstdlib>
+#include <mutex>
+#include <optional>
+
 namespace cudf {
 
 namespace {