Skip to content

Commit b153aac

Browse files
committed
request prealloc_y sync after quantization
1 parent d2f8f00 commit b153aac

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ggml/src/ggml-vulkan/ggml-vulkan.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7493,7 +7493,7 @@ static void ggml_vk_mul_mat_id_q_f16(ggml_backend_vk_context * ctx, vk_context&
74937493
if (x_non_contig || qx_needs_dequant) {
74947494
ctx->prealloc_x_need_sync = true;
74957495
}
7496-
if (y_non_contig) {
7496+
if (y_non_contig || quantize_y) {
74977497
ctx->prealloc_y_need_sync = true;
74987498
}
74997499
}
@@ -7793,7 +7793,7 @@ static void ggml_vk_mul_mat_vec_id_q_f16(ggml_backend_vk_context * ctx, vk_conte
77937793
if (x_non_contig) {
77947794
ctx->prealloc_x_need_sync = true;
77957795
}
7796-
if (y_non_contig) {
7796+
if (y_non_contig || quantize_y) {
77977797
ctx->prealloc_y_need_sync = true;
77987798
}
77997799
}

0 commit comments

Comments
 (0)