NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-03-14 20:27:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Feedback

Browse Source
This commit is contained in:
Bernhard Manfred Gruber
2026-02-28 01:19:18 +01:00
parent 0abc8ec82b
commit 4164909c52
1 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
python/scripts/nvbench_compare.py
View File

@@ -79,6 +79,12 @@ def format_axis_value(axis_name, axis_value, axes):
return format_string_axis_value(axis_name, axis_value, axes)
def make_display(name: str, display_values: [list[str]]) -> str:
open_bracket, close_bracket = ("[", "]") if len(display_values) > 1 else ("", "")
display_values = ",".join(display_values)
return f"{name}={open_bracket}{display_values}{close_bracket}"
def parse_axis_filters(axis_args):
filters = []
for axis_arg in axis_args:
@@ -91,7 +97,6 @@ def parse_axis_filters(axis_args):
raise ValueError("Axis filter must be NAME=VALUE: {}".format(axis_arg))
values = []
display_values = []
if value.startswith("[") and value.endswith("]"):
inner = value[1:-1].strip()
values = [
@@ -121,10 +126,7 @@ def parse_axis_filters(axis_args):
"Axis filter must specify at least one value: {}".format(axis_arg)
)
if len(display_values) == 1:
display = "{}={}".format(name, display_values[0])
else:
display = "{}=[{}]".format(name, ",".join(display_values))
display = make_display(name, display_values)
filters.append(
{
"name": name,