Print timings in sweep-bench (#1454)

This commit is contained in:
Nexes the Elder 2026-03-18 06:57:00 +01:00 committed by GitHub
parent a399456c12
commit 61fad8b094
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 0 deletions

View File

@ -136,6 +136,8 @@ int main(int argc, char ** argv) {
common_batch_clear(batch);
llama_kv_cache_clear(ctx);
llama_reset_timings(ctx);
int i_loop = 0;
for (unsigned int n_kv = 0; n_kv < n_kv_max; n_kv += params.n_ubatch) {
@ -211,6 +213,8 @@ int main(int argc, char ** argv) {
++i_loop;
}
llama_print_timings(ctx);
llama_batch_free(batch);
llama_free(ctx);