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

Update compare script and test files for new JSON.

Browse Source
This commit is contained in:
Allison Vacanti
2022-01-11 17:41:04 -05:00
parent 348acbd6eb
commit 8ba2cf1395
3 changed files with 31112 additions and 33658 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
scripts/nvbench_compare.py
View File

@@ -112,6 +112,8 @@ def compare_benches(ref_benches, cmp_benches, threshold):
ref_states = ref_bench["states"]
cmp_states = cmp_bench["states"]
axes = axes if axes else []
headers = [x["name"] for x in axes]
colalign = ["center"] * len(headers)
@@ -281,10 +283,8 @@ def main():
for ref, comp in to_compare:
with open(ref, "r") as ref_file:
ref_root = reader.read_file(ref_file)
with open(comp, "r") as cmp_file:
cmp_root = reader.read_file(cmp_file)
ref_root = reader.read_file(ref)
cmp_root = reader.read_file(comp)
global all_devices
all_devices = cmp_root["devices"]