Skip to content

Commit a82c9e7

Browse files
authored
vulkan: fix assertion when qy_needs_dequant (#12068)
Looks like a copy/paste bug from qx_needs_dequant.
1 parent 401af80 commit a82c9e7

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
@@ -4192,7 +4192,7 @@ static void ggml_vk_mul_mat_q_f16(ggml_backend_vk_context * ctx, vk_context& sub
41924192
}
41934193
if (qy_needs_dequant) {
41944194
d_Y = ctx->prealloc_y;
4195-
GGML_ASSERT(d_Y->size >= y_sz * ne02 * ne03);
4195+
GGML_ASSERT(d_Y->size >= y_sz * ne12 * ne13);
41964196
} else {
41974197
d_Y = d_Qy;
41984198
y_buf_offset = qy_buf_offset;
@@ -4769,7 +4769,7 @@ static void ggml_vk_mul_mat_id_q_f16(ggml_backend_vk_context * ctx, vk_context&
47694769
}
47704770
if (qy_needs_dequant) {
47714771
d_Y = ctx->prealloc_y;
4772-
GGML_ASSERT(d_Y->size >= y_sz * ne02 * ne03);
4772+
GGML_ASSERT(d_Y->size >= y_sz * ne12 * ne13);
47734773
} else {
47744774
d_Y = d_Qy;
47754775
y_buf_offset = qy_buf_offset;

0 commit comments

Comments
 (0)