Skip to content

Commit

Permalink
Fixing up python script
Browse files Browse the repository at this point in the history
  • Loading branch information
JosephBond committed Oct 24, 2023
1 parent a4bf641 commit c095dea
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions plot_bench.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@

bench_sets = {
'all': ['Trace-Eval', 'Trace-Bwd', 'Trace-Fwd', 'Graph-Eval', 'Graph-Bwd', 'Graph-BwdDual', 'Graph-BwdAll', 'Graph-Fwd', 'Graph-FwdDual', 'Graph-FwdAsDeMorgan'],
'bwd': ['Trace-Eval','Trace-Bwd', 'Graph-Eval', 'Graph-Bwd'],
'bwd': ['Trace-Eval','Trace-Bwd','Graph-Eval', 'Graph-Bwd'],
'fwd': ['Trace-Eval', 'Trace-Fwd', 'Graph-Eval', 'Graph-Fwd', 'Graph-FwdAsDeMorgan'],
'standard': ['Trace-Eval','Trace-Bwd', 'Trace-Fwd', 'Graph-Eval', 'Graph-Bwd', 'Graph-Fwd'],
}
Expand Down Expand Up @@ -73,6 +73,6 @@ def bench_names(bench_str):
capt = "Tests: " + args.Tests + ", Benches: " + args.Benches
lab = args.Tests + '-' + args.Benches
if args.Dest:
parse(tests, benches, capt, lab, args.Dest)
parse(tests, benches, capt, lab, dest=args.Dest)
else:
parse(tests, benches, capt, lab)

0 comments on commit c095dea

Please sign in to comment.