Skip to content

Commit c7b4e7c

Browse files
committed
Add scale-test --save-temps option.
I'm not sure how anyone debugs these tests otherwise.
1 parent 6befb10 commit c7b4e7c

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

utils/scale-test

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -178,7 +178,8 @@ def run_once_with_primary(args, ast, rng, primary_idx):
178178
else:
179179
r = merge_all_jobstats(load_stats_dir(d)).stats
180180
finally:
181-
shutil.rmtree(d)
181+
if not args.save_temps:
182+
shutil.rmtree(d)
182183

183184
return {k: v for (k, v) in r.items() if args.select in k}
184185

@@ -754,6 +755,9 @@ def main():
754755
parser.add_argument(
755756
'--tmpdir', type=str,
756757
default=None, help='directory to create tempfiles in')
758+
parser.add_argument(
759+
'--save-temps', action='store_true',
760+
default=False, help='save files in tempfiles')
757761
parser.add_argument(
758762
'--select',
759763
default="", help='substring of counters/symbols to limit attention to')

0 commit comments

Comments
 (0)