root/llama.cpp
1
0
Fork 0
mirror of https://github.com/ggerganov/llama.cpp.git synced 2025-01-06 00:34:35 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

llama-bench : skip repeated values in consecutive lines

Browse Source
This commit is contained in:
slaren 2024-11-02 15:37:33 +01:00
parent 1926d6e39d
commit 4fc8673d09
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1095,6 +1095,8 @@ struct jsonl_printer : public printer {
struct markdown_printer : public printer {
std::vector<std::string> fields;
std::vector<std::string> prev_values;
static int get_field_width(const std::string & field) {
if (field == "model") {
return -30;
@ -1242,8 +1244,10 @@ struct markdown_printer : public printer {
void print_test(const test & t) override {
std::map<std::string, std::string> vmap = t.get_map();
std::vector<std::string> values;
fprintf(fout, "|");
size_t i = 0;
for (const auto & field : fields) {
std::string value;
char buf[128];
@ -1292,9 +1296,20 @@ struct markdown_printer : public printer {
// HACK: the utf-8 character is 2 bytes
width += 1;
}
values.push_back(value);
if (prev_values.size() > i && prev_values.at(i) == value) {
value = "";
}
fprintf(fout, " %*s |", width, value.c_str());
i++;
}
fprintf(fout, "\n");
prev_values = std::move(values);
}
void print_footer() override {