diff options
Diffstat (limited to 'ggml-cuda')
-rw-r--r-- | ggml-cuda/common.cuh | 7 | ||||
-rw-r--r-- | ggml-cuda/dmmv.cu | 8 | ||||
-rw-r--r-- | ggml-cuda/dmmv.cuh | 11 |
3 files changed, 15 insertions, 11 deletions
diff --git a/ggml-cuda/common.cuh b/ggml-cuda/common.cuh index 33c8ed1d..b98d7cbd 100644 --- a/ggml-cuda/common.cuh +++ b/ggml-cuda/common.cuh @@ -1,7 +1,8 @@ #pragma once -#include "../ggml.h" -#include "../ggml-cuda.h" +#include "ggml.h" +#include "ggml-cuda.h" + #include <memory> #if defined(GGML_USE_HIPBLAS) @@ -11,7 +12,7 @@ #define GGML_COMMON_DECL_CUDA #define GGML_COMMON_IMPL_CUDA #endif -#include "../ggml-common.h" +#include "ggml-common.h" #include <cstdio> #include <array> diff --git a/ggml-cuda/dmmv.cu b/ggml-cuda/dmmv.cu index 7156c921..0b17e3cb 100644 --- a/ggml-cuda/dmmv.cu +++ b/ggml-cuda/dmmv.cu @@ -2,14 +2,6 @@ #include "dequantize.cuh" #include "convert.cuh" -// dmmv = dequantize_mul_mat_vec -#ifndef GGML_CUDA_DMMV_X -#define GGML_CUDA_DMMV_X 32 -#endif -#ifndef GGML_CUDA_MMV_Y -#define GGML_CUDA_MMV_Y 1 -#endif - #ifndef K_QUANTS_PER_ITERATION #define K_QUANTS_PER_ITERATION 2 #else diff --git a/ggml-cuda/dmmv.cuh b/ggml-cuda/dmmv.cuh index 3802678f..4c5ebd47 100644 --- a/ggml-cuda/dmmv.cuh +++ b/ggml-cuda/dmmv.cuh @@ -1,5 +1,16 @@ #include "common.cuh" +// dmmv = dequantize_mul_mat_vec + +// TODO: remove this? +#ifndef GGML_CUDA_DMMV_X +#define GGML_CUDA_DMMV_X 32 +#endif + +#ifndef GGML_CUDA_MMV_Y +#define GGML_CUDA_MMV_Y 1 +#endif + void ggml_cuda_op_dequantize_mul_mat_vec( ggml_backend_cuda_context & ctx, const ggml_tensor * src0, const ggml_tensor * src1, ggml_tensor * dst, const char * src0_dd_i, const float * src1_ddf_i, |