increase default blas batch size

This commit is contained in:
Concedo 2023-04-09 15:27:43 +08:00
parent 4d1825263b
commit b91abc3316
3 changed files with 3 additions and 2 deletions

View file

@ -158,7 +158,7 @@ generation_outputs llama_generate(const generation_inputs inputs, generation_out
int original_threads = params.n_threads;
if (blasmode)
{
params.n_batch = 512;
params.n_batch = 1024;
params.n_threads = 1;
}