Skip to content

Commit

Permalink
scripts : fix missing key in compare-llama-bench.py (#10332)
Browse files Browse the repository at this point in the history
  • Loading branch information
ggerganov authored Nov 16, 2024
1 parent 772703c commit f245cc2
Showing 1 changed file with 2 additions and 5 deletions.
7 changes: 2 additions & 5 deletions scripts/compare-llama-bench.py
Original file line number Diff line number Diff line change
Expand Up @@ -303,14 +303,11 @@ def get_rows(properties):

show = []
# Show CPU and/or GPU by default even if the hardware for all results is the same:
if "gpu_blas" not in properties_different and "n_gpu_layers" not in properties_different:
gpu_blas = bool(rows_full[0][KEY_PROPERTIES.index("gpu_blas")])
if "n_gpu_layers" not in properties_different:
ngl = int(rows_full[0][KEY_PROPERTIES.index("n_gpu_layers")])

if not gpu_blas or ngl != 99 and "cpu_info" not in properties_different:
if ngl != 99 and "cpu_info" not in properties_different:
show.append("cpu_info")
if gpu_blas and "gpu_info" not in properties_different:
show.append("gpu_info")

show += properties_different

Expand Down

0 comments on commit f245cc2

Please sign in to comment.