From ed040d585c3237d706973d7ad290bfee40958270 Mon Sep 17 00:00:00 2001
From: Allison Vacanti <alliepiper16@gmail.com>
Date: Wed, 9 Feb 2022 12:22:11 -0500
Subject: [PATCH] Add missing <iterator> headers.

Fixes #435.
---
 .../dispatch/dispatch_adjacent_difference.cuh       | 13 +++++++------
 cub/device/dispatch/dispatch_unique_by_key.cuh      |  9 +++++----
 cub/util_type.cuh                                   |  1 +
 3 files changed, 13 insertions(+), 10 deletions(-)

diff --git a/cub/device/dispatch/dispatch_adjacent_difference.cuh b/cub/device/dispatch/dispatch_adjacent_difference.cuh
index abfcf42d65..77b060c87b 100644
--- a/cub/device/dispatch/dispatch_adjacent_difference.cuh
+++ b/cub/device/dispatch/dispatch_adjacent_difference.cuh
@@ -27,15 +27,16 @@
 
 #pragma once
 
-#include "../../config.cuh"
-#include "../../util_math.cuh"
-#include "../../util_device.cuh"
-#include "../../util_namespace.cuh"
-#include "../../detail/type_traits.cuh"
-#include "../../agent/agent_adjacent_difference.cuh"
+#include <cub/agent/agent_adjacent_difference.cuh>
+#include <cub/config.cuh>
+#include <cub/detail/type_traits.cuh>
+#include <cub/util_device.cuh>
+#include <cub/util_math.cuh>
+#include <cub/util_namespace.cuh>
 
 #include <thrust/system/cuda/detail/core/triple_chevron_launch.h>
 
+#include <iterator>
 
 CUB_NAMESPACE_BEGIN
 
diff --git a/cub/device/dispatch/dispatch_unique_by_key.cuh b/cub/device/dispatch/dispatch_unique_by_key.cuh
index 8597e3c821..7c93ec4e0f 100644
--- a/cub/device/dispatch/dispatch_unique_by_key.cuh
+++ b/cub/device/dispatch/dispatch_unique_by_key.cuh
@@ -31,11 +31,12 @@
  * cub::DeviceSelect::UniqueByKey provides device-wide, parallel operations for selecting unique items by key from sequences of data items residing within device-accessible memory.
  */
 
-#include "../../agent/agent_unique_by_key.cuh"
-#include "../../util_math.cuh"
-#include "../../util_macro.cuh"
+#include <cub/agent/agent_unique_by_key.cuh>
+#include <cub/device/dispatch/dispatch_scan.cuh>
+#include <cub/util_macro.cuh>
+#include <cub/util_math.cuh>
 
-#include "dispatch_scan.cuh"
+#include <iterator>
 
 CUB_NAMESPACE_BEGIN
 
diff --git a/cub/util_type.cuh b/cub/util_type.cuh
index f8205e2015..d859bb16cb 100644
--- a/cub/util_type.cuh
+++ b/cub/util_type.cuh
@@ -35,6 +35,7 @@
 
 #include <cfloat>
 #include <iostream>
+#include <iterator>
 #include <limits>
 #include <type_traits>