From: Oliver Simons <redacted>
Date: Mon, 11 May 2026 10:16:38 +0000 (+0200)
Subject: CUDA: directly include cuda/iterator (#22936)
X-Git-Tag: upstream/0.0.10438~1333
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=8cef8201a1e0213662abbfcbcd3ff2eb773174df;p=pkg%2Fggml%2Fsources%2Fllama.cpp

CUDA: directly include cuda/iterator (#22936)

Before, we relied on a transient import from `cub/cub.cuh`, which is
bad practice to do as cub may not always expose cuda/iterator
---

diff --git a/ggml/src/ggml-cuda/argsort.cu b/ggml/src/ggml-cuda/argsort.cu
index 0f3f017b5..c4f08091e 100644
--- a/ggml/src/ggml-cuda/argsort.cu
+++ b/ggml/src/ggml-cuda/argsort.cu
@@ -4,6 +4,7 @@
 #    include <cub/cub.cuh>
 #    if (CCCL_MAJOR_VERSION >= 3 && CCCL_MINOR_VERSION >= 1)
 #        define STRIDED_ITERATOR_AVAILABLE
+#        include <cuda/iterator>
 #    endif
 using namespace cub;
 #endif  // GGML_CUDA_USE_CUB