Use a custom batch size threshold for using a batch-specific kernel.

MrMage · Nov 11, 2024 · 34d5a0b · 34d5a0b
1 parent 8de1bee
commit 34d5a0b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ggml/src/ggml-metal.m b/ggml/src/ggml-metal.m
@@ -1884,7 +1884,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 = 6;
 
 #if 0
                 // the numbers below are measured on M2 Ultra for 7B and 13B models