Update mul_mat_vec_q6_k.comp

ggerganov · Dec 28, 2024 · b3b30e4 · b3b30e4
1 parent 25d7ae4
commit b3b30e4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_q6_k.comp b/ggml/src/ggml-vulkan/vulkan-shaders/mul_mat_vec_q6_k.comp
@@ -11,7 +11,7 @@ layout (constant_id = 1) const uint NUM_ROWS = 1;
 
 shared FLOAT_TYPE tmpsh[NUM_ROWS][BLOCK_SIZE];
 shared FLOAT_TYPE sccache[BLOCK_SIZE/16][16];
-shared block_q6_K_packed16 blkcache[BLOCK_SIZE/16];
+shared block_q6_K_packed16 blkcache[BLOCK_SIZE/16 + 1];
 
 void compute_outputs(const uint32_t first_row, const uint32_t num_rows) {
     uint a_offset, b_offset, d_offset;