Add defaults to all workload presets if not present

OmerFarukTutkun · May 9, 2024 · 13c1c5b · 13c1c5b
1 parent 1a95253
commit 13c1c5b
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/OpenBench/config.py b/OpenBench/config.py
@@ -65,6 +65,10 @@ def load_engine_config(engine_name):
         verify_engine_basics(conf)
         verify_engine_build(engine_name, conf)
 
+        for preset_type in ['test_presets', 'tune_presets', 'datagen_presets']:
+            if preset_type not in conf.keys() or 'default' not in conf[preset_type].keys():
+                conf[preset_type] = { 'default' : {} }
+
         assert 'default' in conf['test_presets'].keys()
         assert 'default' in conf['tune_presets'].keys()
         assert 'default' in conf['datagen_presets'].keys()