Skip to content

Commit 84c6782

Browse files
authored
CUDA: Continue directly including cuda/iterator (ggml-org#23102)
Cont of ggml-org#22936, forgot to update one site
1 parent 3e12fbd commit 84c6782

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

ggml/src/ggml-cuda/top-k.cu

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
# include <cub/cub.cuh>
66
# if (CCCL_MAJOR_VERSION >= 3 && CCCL_MINOR_VERSION >= 2)
77
# define CUB_TOP_K_AVAILABLE
8+
# include <cuda/iterator>
89
using namespace cub;
910
# endif // CCCL_MAJOR_VERSION >= 3 && CCCL_MINOR_VERSION >= 2
1011
#endif // GGML_CUDA_USE_CUB

0 commit comments

Comments
 (0)