root/llama.cpp
1
0
Fork 0
mirror of https://github.com/ggerganov/llama.cpp.git synced 2024-11-11 13:30:35 +00:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

llama-bench : add time-to-first-byte stat

Browse Source
This commit is contained in:
Georgi Gerganov 2024-09-19 09:15:29 +03:00
parent 64c6af3195
commit ff231de553
No known key found for this signature in database
GPG Key ID: 449E073F9DC10735
1 changed files with 23 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
examples/llama-bench/llama-bench.cpp
View File

@ -931,6 +931,13 @@ struct test {
return ts;
}
std::vector<double> get_ttfb() const {
int n_tokens = n_prompt + n_gen;
std::vector<double> ts;
std::transform(samples_ns.begin(), samples_ns.end(), std::back_inserter(ts), [n_tokens](uint64_t t) { return t/1e6; });
return ts;
}
double avg_ts() const {
return ::avg(get_ts());
}
@ -939,6 +946,14 @@ struct test {
return ::stdev(get_ts());
}
double avg_ttfb() const {
return ::avg(get_ttfb());
}
double stdev_ttfb() const {
return ::stdev(get_ttfb());
}
static std::string get_backend() {
if (cuda) {
return GGML_CUDA_NAME;
@ -1189,6 +1204,9 @@ struct markdown_printer : public printer {
if (field == "model") {
return -30;
}
if (field == "ttfb") {
return 30;
}
if (field == "t/s") {
return 20;
}
@ -1316,6 +1334,7 @@ struct markdown_printer : public printer {
}
fields.emplace_back("test");
fields.emplace_back("t/s");
fields.emplace_back("ttfb");
fprintf(fout, "|");
for (const auto & field : fields) {
@ -1370,6 +1389,9 @@ struct markdown_printer : public printer {
} else if (field == "t/s") {
snprintf(buf, sizeof(buf), "%.2f ± %.2f", t.avg_ts(), t.stdev_ts());
value = buf;
} else if (field == "ttfb") {
snprintf(buf, sizeof(buf), "%.2f ± %.2f", t.avg_ttfb(), t.stdev_ttfb());
value = buf;
} else if (vmap.find(field) != vmap.end()) {
value = vmap.at(field);
} else {
@ -1378,7 +1400,7 @@ struct markdown_printer : public printer {
}
int width = get_field_width(field);
if (field == "t/s") {
if (field == "t/s" || field == "ttfb") {
// HACK: the utf-8 character is 2 bytes
width += 1;
}