diff options
| author | Florian Fischer <florian.fl.fischer@fau.de> | 2019-08-26 20:34:43 +0200 |
|---|---|---|
| committer | Florian Fischer <florian.fl.fischer@fau.de> | 2019-08-26 20:34:43 +0200 |
| commit | 5d106be3ed2cfae5da5a5d87e7826b7c45870f09 (patch) | |
| tree | 18a71e62629eb4f947be270e3a7522dd22827212 /scripts/merge_saves.py | |
| parent | 18f850dd06b771aa625f3f2f8a4a0a9efc433cd2 (diff) | |
| download | allocbench-5d106be3ed2cfae5da5a5d87e7826b7c45870f09.tar.gz allocbench-5d106be3ed2cfae5da5a5d87e7826b7c45870f09.zip | |
move scripts from scripts/ to root and fix some pylint hints
Diffstat (limited to 'scripts/merge_saves.py')
| -rwxr-xr-x | scripts/merge_saves.py | 61 |
1 files changed, 0 insertions, 61 deletions
diff --git a/scripts/merge_saves.py b/scripts/merge_saves.py deleted file mode 100755 index 7599a1e..0000000 --- a/scripts/merge_saves.py +++ /dev/null @@ -1,61 +0,0 @@ -#!/usr/bin/env python3 - -import argparse -import inspect -import os -import pickle -import sys - -currentdir = os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe()))) -parentdir = os.path.dirname(currentdir) -sys.path.insert(0, parentdir) - -parser = argparse.ArgumentParser(description="Summarize allocbench results in allocator sets") -parser.add_argument("src", help="path to results which should be merged into dest", type=str) -parser.add_argument("dest", help="path to results in which src should be merged", type=str) -parser.add_argument("-b", "--benchmarks", help="benchmarks to summarize", nargs='+') -parser.add_argument("-x", "--exclude-benchmarks", help="benchmarks to exclude", nargs='+') - - -def main(): - args = parser.parse_args() - - for src_save in os.listdir(args.src): - if not src_save.endswith(".save"): - continue - if src_save == "facts.save": - continue - if args.benchmarks and not src_save[:-5] in args.benchmarks: - continue - if args.exclude_benchmarks and src_save[:-5] in args.exclude_benchmarks: - continue - - src_save = os.path.join(args.src, src_save) - dest_save = os.path.join(args.dest, os.path.basename(src_save)) - - if not os.path.isfile(dest_save): - print("Can't merge", src_save, "because", os.path.basename(src_save), "not in", args.dest) - continue - - with open(src_save, "rb") as f: - src_results = pickle.load(f) - - with open(dest_save, "rb") as f: - dest_results = pickle.load(f) - - for alloc in src_results["allocators"]: - if alloc in dest_results["allocators"]: - print(alloc, "already in", dest_save) - continue - - print("merging", alloc, "from", src_save, "into", dest_save) - dest_results["allocators"][alloc] = src_results["allocators"][alloc] - dest_results[alloc] = src_results[alloc] - dest_results["stats"][alloc] = src_results["stats"][alloc] - - with open(dest_save, "wb") as f: - pickle.dump(dest_results, f) - - -if __name__ == "__main__": - main() |
