vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix bug in Vulkan CPY op

Browse source
This commit is contained in:
0cc4m 2024-01-30 19:57:10 +01:00
parent 3c9b68df23
commit ffe841cdd9
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
ggml-vulkan.cpp
View file

@ -2873,7 +2873,8 @@ static void ggml_vk_op_f32(vk_context * ctx, const ggml_tensor * src0, const ggm
if (op == GGML_OP_CPY) { if (op == GGML_OP_CPY) {
GGML_ASSERT(!transfer_src0); GGML_ASSERT(!transfer_src0);
GGML_ASSERT(!transfer_src1); GGML_ASSERT(!transfer_src1);
d_sz = dst->ne[1] * dst->nb[1]; x_sz = ggml_nbytes(src0);
d_sz = ggml_nbytes(dst);
if (extra->offset + d_sz >= d_D->size) { if (extra->offset + d_sz >= d_D->size) {
d_sz = VK_WHOLE_SIZE; d_sz = VK_WHOLE_SIZE;