From 3df67ab3d75b08c72abe86159ac30eccf3ba5d10 Mon Sep 17 00:00:00 2001
From: Bernhard Manfred Gruber <bernhardmgruber@gmail.com>
Date: Tue, 2 Jul 2024 10:34:39 +0200
Subject: [PATCH] Fix include for <iterator>

Follow-up fix for PR #1863
---
 cub/cub/util_type.cuh | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/cub/cub/util_type.cuh b/cub/cub/util_type.cuh
index 5afd2dd23a..30c228c633 100644
--- a/cub/cub/util_type.cuh
+++ b/cub/cub/util_type.cuh
@@ -46,7 +46,6 @@
 #include <cub/detail/uninitialized_copy.cuh>
 
 #include <cuda/std/cstdint>
-#include <cuda/std/iterator>
 #include <cuda/std/limits>
 #include <cuda/std/type_traits>
 
@@ -59,6 +58,12 @@ _CCCL_DIAG_POP
 #  endif // !_CCCL_CUDACC_BELOW_11_8
 #endif // _CCCL_HAS_NV_BF16
 
+#ifdef _CCCL_COMPILER_NVRTC
+#  include <cuda/std/iterator>
+#else // !defined(_CCCL_COMPILER_NVRTC)
+#  include <iterator>
+#endif // defined(_CCCL_COMPILER_NVRTC)
+
 CUB_NAMESPACE_BEGIN
 
 #ifndef CUB_IS_INT128_ENABLED