Polybench: allow passing dataset size

tum-ei-eda · Sep 27, 2024 · 3a15a04 · 3a15a04
1 parent 9efdcb4
commit 3a15a04
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/mlonmcu/models/frontend.py b/mlonmcu/models/frontend.py
@@ -1697,6 +1697,12 @@ def get_platform_config(self, platform):
 
 
 class PolybenchFrontend(SimpleFrontend):
+
+    DEFAULTS = {
+        **Frontend.DEFAULTS,
+        "dataset": "large",  # mini/small/medium/large/extralarge
+    }
+
     REQUIRED = {"polybench.src_dir"}
 
     def __init__(self, features=None, config=None):
@@ -1769,6 +1775,7 @@ def get_platform_defs(self, platform):
         ret = {}
         if platform == "mlif":
             ret["POLYBENCH_DIR"] = Path(self.config["polybench.src_dir"])
+            ret["POLYBENCH_DATASET"] = self.config["dataset"].upper() + "_DATASET"
         return ret
 
     def get_platform_config(self, platform):