From d696eef8711c7054beaac65e43ca24398a4ea677 Mon Sep 17 00:00:00 2001
From: aamijar <aamijar230@gmail.com>
Date: Thu, 19 Feb 2026 12:43:53 -0800
Subject: [PATCH 1/3] fix-thrust-header-2

---
 cpp/src/neighbors/detail/knn_brute_force.cuh | 1 +
 1 file changed, 1 insertion(+)

diff --git a/cpp/src/neighbors/detail/knn_brute_force.cuh b/cpp/src/neighbors/detail/knn_brute_force.cuh
index da9a2e78d7..b3302fbf4f 100644
--- a/cpp/src/neighbors/detail/knn_brute_force.cuh
+++ b/cpp/src/neighbors/detail/knn_brute_force.cuh
@@ -40,6 +40,7 @@
 #include <cuda_fp16.h>
 #include <rmm/cuda_device.hpp>
 #include <rmm/device_uvector.hpp>
+#include <thrust/iterator/counting_iterator.h>
 #include <thrust/iterator/transform_iterator.h>
 
 #include <cstdint>

From c2676fcd8d6c7a7e43944a904a1908c9abfc328d Mon Sep 17 00:00:00 2001
From: Anupam <54245698+aamijar@users.noreply.github.com>
Date: Thu, 19 Feb 2026 13:27:33 -0800
Subject: [PATCH 2/3] Update cpp/src/neighbors/detail/knn_brute_force.cuh

Co-authored-by: Bradley Dice <bdice@bradleydice.com>
---
 cpp/src/neighbors/detail/knn_brute_force.cuh | 1 +
 1 file changed, 1 insertion(+)

diff --git a/cpp/src/neighbors/detail/knn_brute_force.cuh b/cpp/src/neighbors/detail/knn_brute_force.cuh
index b3302fbf4f..5aac16af95 100644
--- a/cpp/src/neighbors/detail/knn_brute_force.cuh
+++ b/cpp/src/neighbors/detail/knn_brute_force.cuh
@@ -40,6 +40,7 @@
 #include <cuda_fp16.h>
 #include <rmm/cuda_device.hpp>
 #include <rmm/device_uvector.hpp>
+#include <thrust/for_each.h>
 #include <thrust/iterator/counting_iterator.h>
 #include <thrust/iterator/transform_iterator.h>
 

From 4b76753baa840f3c0ca3e39b519cbadfaf7b0332 Mon Sep 17 00:00:00 2001
From: aamijar <aamijar230@gmail.com>
Date: Thu, 19 Feb 2026 13:28:14 -0800
Subject: [PATCH 3/3] remove unused transform_iterator.h

---
 cpp/src/neighbors/detail/knn_brute_force.cuh | 1 -
 1 file changed, 1 deletion(-)

diff --git a/cpp/src/neighbors/detail/knn_brute_force.cuh b/cpp/src/neighbors/detail/knn_brute_force.cuh
index 5aac16af95..a226e1d0cf 100644
--- a/cpp/src/neighbors/detail/knn_brute_force.cuh
+++ b/cpp/src/neighbors/detail/knn_brute_force.cuh
@@ -42,7 +42,6 @@
 #include <rmm/device_uvector.hpp>
 #include <thrust/for_each.h>
 #include <thrust/iterator/counting_iterator.h>
-#include <thrust/iterator/transform_iterator.h>
 
 #include <cstdint>
 #include <iostream>