CUDA: Continue directly including cuda/iterator (llama/23102)

Cont of #22936, forgot to update one site
2026-05-17 18:00:10 +02:00 · 2026-05-17 18:00:10 +02:00 · 53736a3f0e
parent 9e96e0eaf1
commit 53736a3f0e
1 changed files with 1 additions and 0 deletions
--- a/ggml/src/ggml-cuda/top-k.cu
+++ b/ggml/src/ggml-cuda/top-k.cu
@ -5,6 +5,7 @@
 #    include <cub/cub.cuh>
 #    if (CCCL_MAJOR_VERSION >= 3 && CCCL_MINOR_VERSION >= 2)
 #        define CUB_TOP_K_AVAILABLE
+#        include <cuda/iterator>
 using namespace cub;
 #    endif  // CCCL_MAJOR_VERSION >= 3 && CCCL_MINOR_VERSION >= 2
 #endif      // GGML_CUDA_USE_CUB