]> git.djapps.eu Git - pkg/ggml/sources/whisper.cpp/commitdiff
metal : enable mat-vec kernels for bs <= 4 (llama/10491)
authorGeorgi Gerganov <redacted>
Mon, 25 Nov 2024 19:49:31 +0000 (21:49 +0200)
committerGeorgi Gerganov <redacted>
Sun, 8 Dec 2024 18:14:35 +0000 (20:14 +0200)
ggml/src/ggml-metal/ggml-metal.m

index ce7afc61c7c21b89a2bbd96f37651b757c63e75d..ae29b8b2e64a05a5527381b037241c965f89e233 100644 (file)
@@ -1935,7 +1935,7 @@ static void ggml_metal_encode_node(
 
                 // find the break-even point where the matrix-matrix kernel becomes more efficient compared
                 // to the matrix-vector kernel
-                int ne11_mm_min = 1;
+                int ne11_mm_min = 4;
 
 #if 0
                 // the numbers below are measured on M2 Ultra for 7B and 13B models