blob: 0eb89a51b7f17b587eee7d73e46392d5bbb2178a (
plain)
1
2
3
4
5
6
7
8
9
10
|
--- a/external/ggml/src/ggml-cuda/top-k.cu
+++ b/external/ggml/src/ggml-cuda/top-k.cu
@@ -4,6 +4,7 @@
#ifdef GGML_CUDA_USE_CUB
# 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
|