Skip to content

Commit 7f2c1d1

Browse files
Fix include for <iterator> (#1929)
Follow-up fix for PR #1863
1 parent 327420b commit 7f2c1d1

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

cub/cub/util_type.cuh

+6-1
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,6 @@
4646
#include <cub/detail/uninitialized_copy.cuh>
4747

4848
#include <cuda/std/cstdint>
49-
#include <cuda/std/iterator>
5049
#include <cuda/std/limits>
5150
#include <cuda/std/type_traits>
5251

@@ -59,6 +58,12 @@ _CCCL_DIAG_POP
5958
# endif // !_CCCL_CUDACC_BELOW_11_8
6059
#endif // _CCCL_HAS_NV_BF16
6160

61+
#ifdef _CCCL_COMPILER_NVRTC
62+
# include <cuda/std/iterator>
63+
#else // !defined(_CCCL_COMPILER_NVRTC)
64+
# include <iterator>
65+
#endif // defined(_CCCL_COMPILER_NVRTC)
66+
6267
CUB_NAMESPACE_BEGIN
6368

6469
#ifndef CUB_IS_INT128_ENABLED

0 commit comments

Comments
 (0)