CS348Project/llama.cpp
5
0
Fork 0
mirror of https://github.com/ggml-org/llama.cpp.git synced 2025-10-30 08:42:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

compare-llama-bench.py: add missing basicConfig (#7138)

Browse Source 
* compare-llama-bench.py: add missing basicConfig

* compare-llama-bench.py: Add line break between error message and print_help()

* Add regular print() markdown table
This commit is contained in:
Brian
2024-05-08 18:54:39 +10:00
committed by GitHub
parent 3855416027
commit acdce3cdef
1 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
scripts/compare-llama-bench.py
View File

@@ -93,11 +93,14 @@ help_s = (
"specified values are averaged WITHOUT weighing by the --repetitions parameter of llama-bench." "specified values are averaged WITHOUT weighing by the --repetitions parameter of llama-bench."
) )
parser.add_argument("-s", "--show", help=help_s) parser.add_argument("-s", "--show", help=help_s)
parser.add_argument("--verbose", action="store_true", help="increase output verbosity")
known_args, unknown_args = parser.parse_known_args() known_args, unknown_args = parser.parse_known_args()
logging.basicConfig(level=logging.DEBUG if known_args.verbose else logging.INFO)
if unknown_args: if unknown_args:
logger.error(f"Received unknown args: {unknown_args}.") logger.error(f"Received unknown args: {unknown_args}.\n")
parser.print_help() parser.print_help()
sys.exit(1) sys.exit(1)
@@ -110,7 +113,7 @@ if input_file is None:
input_file = sqlite_files[0] input_file = sqlite_files[0]
if input_file is None: if input_file is None:
logger.error("Cannot find a suitable input file, please provide one.") logger.error("Cannot find a suitable input file, please provide one.\n")
parser.print_help() parser.print_help()
sys.exit(1) sys.exit(1)
@@ -202,12 +205,12 @@ elif repo is not None:
hexsha8_baseline = find_parent_in_data(repo.heads.master.commit) hexsha8_baseline = find_parent_in_data(repo.heads.master.commit)
if hexsha8_baseline is None: if hexsha8_baseline is None:
logger.error("No baseline was provided and did not find data for any master branch commits.") logger.error("No baseline was provided and did not find data for any master branch commits.\n")
parser.print_help() parser.print_help()
sys.exit(1) sys.exit(1)
else: else:
logger.error("No baseline was provided and the current working directory " logger.error("No baseline was provided and the current working directory "
"is not part of a git repository from which a baseline could be inferred.") "is not part of a git repository from which a baseline could be inferred.\n")
parser.print_help() parser.print_help()
sys.exit(1) sys.exit(1)
@@ -238,7 +241,7 @@ elif repo is not None:
break break
if hexsha8_compare is None: if hexsha8_compare is None:
logger.error("No compare target was provided and did not find data for any non-master commits.") logger.error("No compare target was provided and did not find data for any non-master commits.\n")
parser.print_help() parser.print_help()
sys.exit(1) sys.exit(1)
else: else:
@@ -361,7 +364,7 @@ if "gpu_info" in show:
headers = [PRETTY_NAMES[p] for p in show] headers = [PRETTY_NAMES[p] for p in show]
headers += ["Test", f"t/s {name_baseline}", f"t/s {name_compare}", "Speedup"] headers += ["Test", f"t/s {name_baseline}", f"t/s {name_compare}", "Speedup"]
logger.info(tabulate( print(tabulate( # noqa: NP100
table, table,
headers=headers, headers=headers,
floatfmt=".2f", floatfmt=".2f",