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

Cont of #22936, forgot to update one site
This commit is contained in:
Oliver Simons 2026-05-17 18:00:10 +02:00 committed by GitHub
parent 3e12fbdea5
commit 84c678242a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -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