Re-enable output-dir option

Author: ianayl

devops/scripts/benchmarks/main.py

@@ -413,6 +413,12 @@ def validate_and_parse_env_args(env_args):
     parser.add_argument(
         "--output-html", help="Create HTML output", action="store_true", default=False
     )
+    parser.add_argument(
+        "--output-dir",
+        type=str,
+        help="Location for output files, if --output-html or --output_markdown was specified.",
+        default=None,
+    )
     parser.add_argument(
         "--dry-run",
         help="Do not run any actual benchmarks",
@@ -500,6 +506,11 @@ def validate_and_parse_env_args(env_args):
     if args.compute_runtime is not None:
         options.build_compute_runtime = True
         options.compute_runtime_tag = args.compute_runtime
+    if args.output_dir is not None:
+        if not os.path.isdir(args.output_dir):
+            parser.error("Specified --output-dir is not a valid path")
+        options.output_directory = os.path.abspath(args.output_dir)
+
 
     benchmark_filter = re.compile(args.filter) if args.filter else None
 

devops/scripts/benchmarks/options.py

@@ -31,6 +31,7 @@ class Options:
     compare_max: int = 10  # average/median over how many results
     output_markdown: MarkdownSize = MarkdownSize.SHORT
     output_html: bool = False
+    output_directory: str = None
     dry_run: bool = False
     stddev_threshold: float = 0.02
     iterations_stddev: int = 5