mlonmcu 0.5.dev0__tar.gz → 0.6.dev0__tar.gz

{mlonmcu-0.5.dev0 → mlonmcu-0.6.dev0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mlonmcu
-Version: 0.5.dev0
+Version: 0.6.dev0
 Summary: This project contains research code related to the deployment of inferenceor learning applications on tiny micro-controllers.
 Home-page: https://github.com/tum-ei-eda/mlonmcu
 Author: TUM Department of Electrical and Computer Engineering - Chair of Electronic Design Automation
@@ -24,6 +24,7 @@ Provides-Extra: espidf
 Provides-Extra: etiss
 Provides-Extra: gvsoc_pulp
 Provides-Extra: microtvm
+Provides-Extra: microtvm_gvsoc
 Provides-Extra: moiopt
 Provides-Extra: onnx
 Provides-Extra: relay-visualization

{mlonmcu-0.5.dev0 → mlonmcu-0.6.dev0}/mlonmcu/feature/feature.py

@@ -229,3 +229,9 @@ class RunFeature(FeatureBase):
 
     def add_run_config(self, config):
         config.update(self.get_run_config())
+
+    # def get_postprocesses(self):
+    #     return []
+
+    # def add_postprocesses(self, postprocesses):
+    #     postprocesses.extend(self.get_postprocesses())

{mlonmcu-0.5.dev0 → mlonmcu-0.6.dev0}/mlonmcu/feature/features.py

@@ -2227,3 +2227,212 @@ class VanillaAccelerator(TargetFeature):
             assert self.name not in extra_plugin_config
             extra_plugin_config[self.name]["baseaddr"] = self.base_addr
             config.update({f"{target}.extra_plugin_config": extra_plugin_config})
+
+
+@register_feature("gen_data")
+class GenData(FrontendFeature):  # TODO: use custom stage instead of LOAD
+    """Generate input data for validation."""
+
+    DEFAULTS = {
+        **FeatureBase.DEFAULTS,
+        "fill_mode": "file",  # Allowed: random, ones, zeros, file, dataset
+        "file": "auto",  # Only relevant if fill_mode=file
+        "number": 10,  # generate up to number samples (may be less if file has less inputs)
+        "fmt": "npy",  # Allowed: npy, npz
+    }
+
+    def __init__(self, features=None, config=None):
+        super().__init__("gen_data", features=features, config=config)
+
+    @property
+    def fill_mode(self):
+        value = self.config["fill_mode"]
+        assert value in ["random", "ones", "zeros", "file", "dataset"]
+        return value
+
+    @property
+    def file(self):
+        value = self.config["file"]
+        return value
+
+    @property
+    def number(self):
+        return int(self.config["number"])
+
+    @property
+    def fmt(self):
+        value = self.config["fmt"]
+        assert value in ["npy", "npz"]
+        return value
+
+    def get_frontend_config(self, frontend):
+        assert frontend in ["tflite"]
+        return {
+            f"{frontend}.gen_data": self.enabled,
+            f"{frontend}.gen_data_fill_mode": self.fill_mode,
+            f"{frontend}.gen_data_file": self.file,
+            f"{frontend}.gen_data_number": self.number,
+            f"{frontend}.gen_data_fmt": self.fmt,
+        }
+
+
+@register_feature("gen_ref_data", depends=["gen_data"])
+class GenRefData(FrontendFeature):  # TODO: use custom stage instead of LOAD
+    """Generate reference outputs for validation."""
+
+    DEFAULTS = {
+        **FeatureBase.DEFAULTS,
+        "mode": "file",  # Allowed: file, model
+        "file": "auto",  # Only relevant if mode=file
+        "fmt": "npy",  # Allowed: npy, npz
+    }
+
+    def __init__(self, features=None, config=None):
+        super().__init__("gen_ref_data", features=features, config=config)
+
+    @property
+    def mode(self):
+        value = self.config["mode"]
+        assert value in ["file", "model"]
+        return value
+
+    @property
+    def file(self):
+        value = self.config["file"]
+        return value
+
+    @property
+    def fmt(self):
+        value = self.config["fmt"]
+        assert value in ["npy", "npz"]
+        return value
+
+    def get_frontend_config(self, frontend):
+        assert frontend in ["tflite"]
+        return {
+            f"{frontend}.gen_ref_data": self.enabled,
+            f"{frontend}.gen_ref_data_mode": self.mode,
+            f"{frontend}.gen_ref_data_file": self.file,
+            f"{frontend}.gen_ref_data_fmt": self.fmt,
+        }
+
+
+@register_feature("gen_ref_labels", depends=["gen_data"])
+class GenRefLabels(FrontendFeature):  # TODO: use custom stage instead of LOAD
+    """Generate reference labels for classification."""
+
+    DEFAULTS = {
+        **FeatureBase.DEFAULTS,
+        "mode": "file",  # Allowed: file, model
+        "file": "auto",  # Only relevant if mode=file
+        "fmt": "npy",  # Allowed: npy, npz, txt
+    }
+
+    def __init__(self, features=None, config=None):
+        super().__init__("gen_ref_labels", features=features, config=config)
+
+    @property
+    def mode(self):
+        value = self.config["mode"]
+        assert value in ["file", "model"]
+        return value
+
+    @property
+    def file(self):
+        value = self.config["file"]
+        return value
+
+    @property
+    def fmt(self):
+        value = self.config["fmt"]
+        assert value in ["npy", "npz"]
+        return value
+
+    def get_frontend_config(self, frontend):
+        assert frontend in ["tflite"]
+        return {
+            f"{frontend}.gen_ref_labels": self.enabled,
+            f"{frontend}.gen_ref_labels_mode": self.mode,
+            f"{frontend}.gen_ref_labels_file": self.file,
+            f"{frontend}.gen_ref_labels_fmt": self.fmt,
+        }
+
+
+@register_feature("set_inputs")
+class SetInputs(PlatformFeature):  # TODO: use custom stage instead of LOAD
+    """Apply test inputs to model."""
+
+    DEFAULTS = {
+        **FeatureBase.DEFAULTS,
+        "interface": "auto",  # Allowed: auto, rom, filesystem, stdin, stdin_raw, uart
+    }
+
+    def __init__(self, features=None, config=None):
+        super().__init__("set_inputs", features=features, config=config)
+
+    @property
+    def interface(self):
+        value = self.config["interface"]
+        assert value in ["auto", "rom", "filesystem", "stdin", "stdin_raw", "uart"]
+        return value
+
+    def get_platform_config(self, platform):
+        assert platform in ["mlif", "tvm", "microtvm"]
+        # if tvm/microtvm: allow using --fill-mode provided by tvmc run
+        return {
+            f"{platform}.set_inputs": self.enabled,
+            f"{platform}.set_inputs_interface": self.interface,
+        }
+
+
+@register_feature("get_outputs")
+class GetOutputs(PlatformFeature):  # TODO: use custom stage instead of LOAD
+    """Extract resulting outputs from model."""
+
+    DEFAULTS = {
+        **FeatureBase.DEFAULTS,
+        "interface": "auto",  # Allowed: auto, filesystem, stdout, stdout_raw, uart
+        "fmt": "npy",  # Allowed: npz, npz
+    }
+
+    def __init__(self, features=None, config=None):
+        super().__init__("get_outputs", features=features, config=config)
+
+    @property
+    def interface(self):
+        value = self.config["interface"]
+        assert value in ["auto", "filesystem", "stdout", "stdout_raw", "uart"]
+        return value
+
+    @property
+    def fmt(self):
+        value = self.config["fmt"]
+        assert value in ["npy", "npz"]
+        return value
+
+    def get_platform_config(self, platform):
+        assert platform in ["mlif", "tvm", "microtvm"]
+        return {
+            f"{platform}.get_outputs": self.enabled,
+            f"{platform}.get_outputs_interface": self.interface,
+            f"{platform}.get_outputs_fmt": self.fmt,
+        }
+
+
+@register_feature("validate_new", depends=["gen_data", "gen_ref_data", "set_inputs", "get_outputs"])
+class ValidateNew(RunFeature):
+    """Wrapper feature for enabling all validatioon related features at once."""
+
+    DEFAULTS = {
+        **FeatureBase.DEFAULTS,
+    }
+
+    def __init__(self, features=None, config=None):
+        super().__init__("validate_new", features=features, config=config)
+
+    # def get_postprocesses(self):
+    #     # config = {}
+    #     # from mlonmcu.session.postprocess import ValidateOutputsPostprocess
+    #     # validate_outputs_postprocess = ValidateOutputsPostprocess(features=[], config=config)
+    #     # return [validate_outputs_postprocess]
+    #     return ["validate_outputs"]

{mlonmcu-0.5.dev0 → mlonmcu-0.6.dev0}/mlonmcu/flow/tvm/backend/backend.py

@@ -160,6 +160,7 @@ class TVMBackend(Backend):
             extra = {}
         if isinstance(extra, str):
             import ast
+
             extra = ast.literal_eval(extra)
         assert isinstance(extra, dict)
         return extra

{mlonmcu-0.5.dev0 → mlonmcu-0.6.dev0}/mlonmcu/flow/tvm/backend/tvmc_utils.py

@@ -164,7 +164,7 @@ def get_tvmrt_tvmc_args(runtime="crt", system_lib=True, link_params=True):
     return ret
 
 
-def get_data_tvmc_args(mode=None, ins_file=None, outs_file=None, print_top=10):
+def get_data_tvmc_args(mode=None, ins_file=None, outs_file=None, print_top=None):
     ret = []
     if ins_file is not None:
         ret.extend(["--inputs", ins_file])
@@ -176,7 +176,7 @@ def get_data_tvmc_args(mode=None, ins_file=None, outs_file=None, print_top=10):
         ret.extend(["--outputs", outs_file])
 
     if print_top is not None and print_top > 0:
-        ret.extend(["--print-top", print_top])
+        ret.extend(["--print-top", str(print_top)])
 
     return ret
 

{mlonmcu-0.5.dev0 → mlonmcu-0.6.dev0}/mlonmcu/models/__init__.py

@@ -33,6 +33,7 @@ from .frontend import (
     MathisFrontend,
     MibenchFrontend,
     LayerGenFrontend,
+    OpenASIPFrontend,
 )
 
 SUPPORTED_FRONTENDS = {
@@ -51,6 +52,7 @@ SUPPORTED_FRONTENDS = {
     "mathis": MathisFrontend,
     "mibench": MibenchFrontend,
     "layergen": LayerGenFrontend,
+    "openasip": OpenASIPFrontend,
 }  # TODO: use registry instead
 
 __all__ = [