nikola/llama_cpp
1
0
Fork 0
Code Issues Pull Requests Actions 22 Packages Projects Releases Wiki Activity

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

Browse Source 
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
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ggml/src/ggml-cuda/top-k.cu
+1
View File
@@ -5,6 +5,7 @@
# include <cub/cub.cuh> # include <cub/cub.cuh>
# if (CCCL_MAJOR_VERSION >= 3 && CCCL_MINOR_VERSION >= 2) # if (CCCL_MAJOR_VERSION >= 3 && CCCL_MINOR_VERSION >= 2)
# define CUB_TOP_K_AVAILABLE # define CUB_TOP_K_AVAILABLE
# include <cuda/iterator>
using namespace cub; using namespace cub;
# endif // CCCL_MAJOR_VERSION >= 3 && CCCL_MINOR_VERSION >= 2 # endif // CCCL_MAJOR_VERSION >= 3 && CCCL_MINOR_VERSION >= 2
#endif // GGML_CUDA_USE_CUB #endif // GGML_CUDA_USE_CUB