NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-05-13 17:55:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

Address code rabbit review feedback

Browse Source
This commit is contained in:
Oleksandr Pavlyk
2026-05-12 15:22:57 -05:00
parent fae9dfca18
commit 151e463fb9
4 changed files with 119 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
python/cuda/bench/results/_benchmark_result.py
View File

@@ -117,14 +117,18 @@ def parse_summary(summary: dict) -> BenchmarkResultSummary:
)
def get_state_summaries(state: dict) -> list[dict]:
return state.get("summaries") or []
def parse_summaries(state: dict) -> dict[str, BenchmarkResultSummary]:
return {
summary["tag"]: parse_summary(summary) for summary in state["summaries"] or []
summary["tag"]: parse_summary(summary) for summary in get_state_summaries(state)
}
def parse_binary_meta(state: dict, tag: str) -> tuple[int | None, str | None]:
summaries = state["summaries"]
summaries = get_state_summaries(state)
if not summaries:
return None, None
@@ -251,7 +255,17 @@ class SubBenchmarkState:
for axis in self.axis_values:
axis_name = axis["name"]
name = axes_names[axis_name]
value = axes_values[axis_name][axis["value"]]
axis_value_map = axes_values[axis_name]
if "value" in axis:
key = str(axis["value"])
value = axis_value_map.get(key, key)
else:
input_string = axis.get("input_string")
value = (
axis_value_map.get(input_string, input_string)
if input_string is not None
else ""
)
self.point[name] = value
def __repr__(self) -> str: