From 3025b621d12a6931ff5e9775d4f644719980ad91 Mon Sep 17 00:00:00 2001 From: R0CKSTAR Date: Sat, 2 Aug 2025 17:20:40 +0800 Subject: [PATCH] llama-bench: rename DB table name from test to llama_bench (#15003) Signed-off-by: Xiaodong Ye --- scripts/compare-llama-bench.py | 10 +++++----- tools/llama-bench/llama-bench.cpp | 4 ++-- 2 files changed, 7 insertions(+), 7 deletions(-) diff --git a/scripts/compare-llama-bench.py b/scripts/compare-llama-bench.py index 406930fb0..c974d83b5 100755 --- a/scripts/compare-llama-bench.py +++ b/scripts/compare-llama-bench.py @@ -326,7 +326,7 @@ class LlamaBenchDataSQLite3(LlamaBenchData): # Set table name and schema based on tool if self.tool == "llama-bench": - self.table_name = "test" + self.table_name = "llama_bench" db_fields = LLAMA_BENCH_DB_FIELDS db_types = LLAMA_BENCH_DB_TYPES elif self.tool == "test-backend-ops": @@ -409,8 +409,8 @@ class LlamaBenchDataSQLite3File(LlamaBenchDataSQLite3): # Tool selection logic if tool is None: - if "test" in table_names: - self.table_name = "test" + if "llama_bench" in table_names: + self.table_name = "llama_bench" self.tool = "llama-bench" elif "test_backend_ops" in table_names: self.table_name = "test_backend_ops" @@ -418,8 +418,8 @@ class LlamaBenchDataSQLite3File(LlamaBenchDataSQLite3): else: raise RuntimeError(f"No suitable table found in database. Available tables: {table_names}") elif tool == "llama-bench": - if "test" in table_names: - self.table_name = "test" + if "llama_bench" in table_names: + self.table_name = "llama_bench" self.tool = "llama-bench" else: raise RuntimeError(f"Table 'test' not found for tool 'llama-bench'. Available tables: {table_names}") diff --git a/tools/llama-bench/llama-bench.cpp b/tools/llama-bench/llama-bench.cpp index c56834a2a..ba0699c43 100644 --- a/tools/llama-bench/llama-bench.cpp +++ b/tools/llama-bench/llama-bench.cpp @@ -1738,7 +1738,7 @@ struct sql_printer : public printer { void print_header(const cmd_params & params) override { std::vector fields = test::get_fields(); - fprintf(fout, "CREATE TABLE IF NOT EXISTS test (\n"); + fprintf(fout, "CREATE TABLE IF NOT EXISTS llama_bench (\n"); for (size_t i = 0; i < fields.size(); i++) { fprintf(fout, " %s %s%s\n", fields.at(i).c_str(), get_sql_field_type(fields.at(i)).c_str(), i < fields.size() - 1 ? "," : ""); @@ -1749,7 +1749,7 @@ struct sql_printer : public printer { } void print_test(const test & t) override { - fprintf(fout, "INSERT INTO test (%s) ", join(test::get_fields(), ", ").c_str()); + fprintf(fout, "INSERT INTO llama_bench (%s) ", join(test::get_fields(), ", ").c_str()); fprintf(fout, "VALUES ("); std::vector values = t.get_values(); for (size_t i = 0; i < values.size(); i++) {