Cherrypick fix from #1825.

mythrocks · mythrocks · commit b59b8c3fb084 · 2026-02-19T16:09:24.000-08:00
Signed-off-by: MithunR &lt;mithunr@nvidia.com&gt;
diff --git a/cpp/src/neighbors/detail/knn_brute_force.cuh b/cpp/src/neighbors/detail/knn_brute_force.cuh
@@ -41,7 +41,7 @@
 #include <rmm/cuda_device.hpp>
 #include <rmm/device_uvector.hpp>
 #include <thrust/for_each.h>
-#include <thrust/iterator/transform_iterator.h>
+#include <thrust/iterator/counting_iterator.h>
 
 #include <cstdint>
 #include <iostream>
