Feat (benchmark): Allow easy specification of different search algorithms within benchmark scripts by nickfraser · Pull Request #1357 · Xilinx/brevitas

src/brevitas_examples/common/benchmark/cfg2randbenchmarkcfg.py

-Original file line number
+Diff line change
@@ -0,0 +1,27 @@
+    # Copyright (C) 2024, Advanced Micro Devices, Inc. All rights reserved.
+    # SPDX-License-Identifier: BSD-3-Clause
+    from argparse import ArgumentParser
+    import yaml
+    if __name__ == "__main__":
+        parser = ArgumentParser()
+        parser.add_argument(
+            '--config',
+            type=str,
+            default=None,
+            required=True,
+            help='Specify a configuration file (YAML) to use as a template for a benchmark config')
+        parser.add_argument(
+            '--benchmark-config',
+            type=str,
+            default=None,
+            required=True,
+            help='Specify the filename of the output benchmark config template')
+        args = parser.parse_args()
+        with open(args.config, 'r') as f:
+            defaults = yaml.safe_load(f)
+        benchmark_defaults = {k: {"rand_type": "const", "rand_values": v} for k, v in defaults.items()}
+        with open(args.benchmark_config, 'w') as f:
+            yaml.dump(benchmark_defaults, f)

Provide feedback