diff options
Diffstat (limited to 'ggml-cuda/convert.cuh')
| -rw-r--r-- | ggml-cuda/convert.cuh | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ggml-cuda/convert.cuh b/ggml-cuda/convert.cuh index db34c0be..5394be9f 100644 --- a/ggml-cuda/convert.cuh +++ b/ggml-cuda/convert.cuh @@ -3,7 +3,7 @@ #define CUDA_DEQUANTIZE_BLOCK_SIZE 256 template<typename T> -using to_t_cuda_t = void (*)(const void * __restrict__ x, T * __restrict__ y, int k, cudaStream_t stream); +using to_t_cuda_t = void (*)(const void * __restrict__ x, T * __restrict__ y, int64_t k, cudaStream_t stream); typedef to_t_cuda_t<float> to_fp32_cuda_t; typedef to_t_cuda_t<half> to_fp16_cuda_t; |