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

Adopt histogram script to new format

Browse Source
This commit is contained in:
Georgy Evtushenko
2022-12-22 17:12:22 +04:00
parent 1cce3988e2
commit 1eb0c49222
1 changed files with 16 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
scripts/nvbench_histogram.py Normal file → Executable file
View File

@@ -38,6 +38,20 @@ def parse_files():
return filenames
def extract_filename(summary):
summary_data = summary["data"]
value_data = next(filter(lambda v: v["name"] == "filename", summary_data))
assert(value_data["type"] == "string")
return value_data["value"]
def extract_size(summary):
summary_data = summary["data"]
value_data = next(filter(lambda v: v["name"] == "size", summary_data))
assert(value_data["type"] == "int64")
return int(value_data["value"])
def parse_samples_meta(filename, state):
summaries = state["summaries"]
if not summaries:
@@ -49,13 +63,13 @@ def parse_samples_meta(filename, state):
if not summary:
return None, None
sample_filename = summary["filename"]["value"]
sample_filename = extract_filename(summary)
# If not absolute, the path is relative to the associated .json file:
if not os.path.isabs(sample_filename):
sample_filename = os.path.join(os.path.dirname(filename), sample_filename)
sample_count = int(summary["size"]["value"])
sample_count = extract_size(summary)
return sample_count, sample_filename