Skip to content

Commit

Permalink
Use a custom batch size threshold for using a batch-specific kernel.
Browse files Browse the repository at this point in the history
  • Loading branch information
MrMage committed Aug 1, 2024
1 parent c8a0090 commit bda44ec
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion ggml/src/ggml-metal.m
Original file line number Diff line number Diff line change
Expand Up @@ -1543,7 +1543,7 @@ static enum ggml_status ggml_metal_graph_compute(

// 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
Expand Down

0 comments on commit bda44ec

Please sign in to comment.