diff options
| author | Wentai Zhang <rchardx@gmail.com> | 2023-09-03 16:46:44 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-09-03 11:46:44 +0300 |
| commit | 6460f758dbd472653296044d36bed8c4554988f5 (patch) | |
| tree | 5cf738cbb1099629adcbe3464b3ad46efa63239e /ggml-opencl.cpp | |
| parent | ca82cf7bac0c91d03e3d320b3a865dd006f854ac (diff) | |
opencl : fix a bug in ggml_cl_pool_malloc() for ggml_cl_mul_mat_f32() (#2955)
Co-authored-by: Wentai Zhang <wentaizhang@tencent.com>
Diffstat (limited to 'ggml-opencl.cpp')
| -rw-r--r-- | ggml-opencl.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ggml-opencl.cpp b/ggml-opencl.cpp index eb214a83..3d50a7f0 100644 --- a/ggml-opencl.cpp +++ b/ggml-opencl.cpp @@ -1493,7 +1493,7 @@ static void ggml_cl_mul_mat_f32(const ggml_tensor * src0, const ggml_tensor * sr if (src0->backend == GGML_BACKEND_GPU) { // NOLINT d_X = (cl_mem) src0->data; } else { - d_X = ggml_cl_pool_malloc(sizeof(ggml_fp16_t) * x_ne, &x_size); + d_X = ggml_cl_pool_malloc(sizeof(float) * x_ne, &x_size); } cl_mem d_Y = ggml_cl_pool_malloc(sizeof(float) * y_ne, &y_size); cl_mem d_D = ggml_cl_pool_malloc(sizeof(float) * d_ne, &d_size); |