diff options
Diffstat (limited to 'ggml-cuda.h')
| -rw-r--r-- | ggml-cuda.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/ggml-cuda.h b/ggml-cuda.h index b105ed0c..1fd67ebe 100644 --- a/ggml-cuda.h +++ b/ggml-cuda.h @@ -1,5 +1,6 @@ #include <cublas_v2.h> #include <cuda_runtime.h> +#include "ggml.h" #ifdef __cplusplus extern "C" { @@ -38,6 +39,8 @@ void dequantize_row_q5_0_cuda(const void * vx, float * y, int k, cudaStream_t st void dequantize_row_q5_1_cuda(const void * vx, float * y, int k, cudaStream_t stream); void dequantize_row_q8_0_cuda(const void * vx, float * y, int k, cudaStream_t stream); +cudaError_t ggml_cuda_h2d_tensor_2d(void * dst, const struct ggml_tensor * src, uint64_t i3, uint64_t i2, cudaStream_t stream); + #ifdef __cplusplus } #endif |