metal : enable mat-vec kernels for bs <= 4 (#10491)

This commit is contained in:
Georgi Gerganov 2024-11-25 21:49:31 +02:00 committed by GitHub
parent 80acb7b430
commit 106964e3d2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

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