Skip to content

Commit dd86df8

Browse files
committed
metal : use mm kernel only for quantum KV cache
1 parent 903167a commit dd86df8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ggml-metal.m

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1129,7 +1129,7 @@ void ggml_metal_graph_compute(
11291129
!ggml_is_transposed(src1) &&
11301130
src1t == GGML_TYPE_F32 &&
11311131
ne00 % 32 == 0 && ne00 >= 64 &&
1132-
(ne11 > ne11_mm_min || ne12 > 1)) {
1132+
(ne11 > ne11_mm_min || (ggml_is_quantized(src0t) && ne12 > 1))) {
11331133
//printf("matrix: ne00 = %6d, ne01 = %6d, ne02 = %6d, ne11 = %6d, ne12 = %6d\n", ne00, ne01, ne02, ne11, ne12);
11341134
switch (src0->type) {
11351135
case GGML_TYPE_F32: [encoder setComputePipelineState:ctx->pipeline_mul_mm_f32_f32]; break;

0 commit comments

Comments
 (0)