mlonmcu 0.7.dev0__tar.gz → 0.8.0__tar.gz

{mlonmcu-0.7.dev0 → mlonmcu-0.8.0}/PKG-INFO

@@ -1,12 +1,13 @@
 Metadata-Version: 2.1
 Name: mlonmcu
-Version: 0.7.dev0
+Version: 0.8.0
 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
 Author-email: philipp.van-kempen@tum.de
 License: Apache License 2.0
 Keywords: mlonmcu
+Platform: UNKNOWN
 Classifier: Development Status :: 2 - Pre-Alpha
 Classifier: Intended Audience :: Developers
 Classifier: License :: OSI Approved :: Apache Software License
@@ -20,6 +21,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Requires-Python: >=3.7
 Description-Content-Type: text/markdown
+Provides-Extra: all-prod
 Provides-Extra: espidf
 Provides-Extra: etiss
 Provides-Extra: gvsoc_pulp
@@ -35,7 +37,6 @@ Provides-Extra: tvm-autotuning
 Provides-Extra: visualize
 Provides-Extra: visualize_tflite
 Provides-Extra: zephyr
-Provides-Extra: all-prod
 License-File: LICENSE
 License-File: NOTICE
 
@@ -189,7 +190,6 @@ with mlonmcu.context.MlonMcuContext() as context:
 Make sure to first install the additonal set of development Python packages into your virtual environment:
 
 ```
-pip install -r requirements_all.txt  # Install packages for every component (instead of using mlonmcu setup -g)
 pip install -r requirements_dev.txt  # Building distributions and running tests
 pip install -r docs/requirements.txt  # For working with the documentation
 ```
@@ -260,3 +260,5 @@ the projects [Scale4Edge](https://www.edacentrum.de/scale4edge/) (grant number 1
 ## 0.1.0 (2021-11-12)
 
 * First closed-source release.
+
+

{mlonmcu-0.7.dev0 → mlonmcu-0.8.0}/README.md

@@ -148,7 +148,6 @@ with mlonmcu.context.MlonMcuContext() as context:
 Make sure to first install the additonal set of development Python packages into your virtual environment:
 
 ```
-pip install -r requirements_all.txt  # Install packages for every component (instead of using mlonmcu setup -g)
 pip install -r requirements_dev.txt  # Building distributions and running tests
 pip install -r docs/requirements.txt  # For working with the documentation
 ```

{mlonmcu-0.7.dev0 → mlonmcu-0.8.0}/mlonmcu/cli/common.py

@@ -167,6 +167,13 @@ def add_flow_options(parser):
         default="",
         help="Label for the session (default: %(default)s)",
     )
+    flow_parser.add_argument(
+        "--dest",
+        type=str,
+        metavar="DEST",
+        default=None,
+        help="Custom destination for session artifacts",
+    )
 
 
 def add_common_options(parser):

{mlonmcu-0.7.dev0 → mlonmcu-0.8.0}/mlonmcu/cli/load.py

@@ -64,7 +64,7 @@ def _handle(args, context):
     config.update(new_config)
     frontends = extract_frontend_names(args, context=context)
     postprocesses = extract_postprocess_names(args, context=context)
-    session = context.get_session(label=args.label, resume=args.resume, config=config)
+    session = context.get_session(label=args.label, resume=args.resume, config=config, dest=args.dest)
     models = apply_modelgroups(args.models, context=context)
     for model in models:
         for f in gen_features:

{mlonmcu-0.7.dev0 → mlonmcu-0.8.0}/mlonmcu/config.py

@@ -18,6 +18,7 @@
 #
 """Collection of utilities to manage MLonMCU configs."""
 import distutils.util
+from typing import List
 import ast
 
 from mlonmcu.feature.type import FeatureType
@@ -244,3 +245,44 @@ def str2list(value, allow_none=False):
     if value.startswith("["):
         return list(ast.literal_eval(value))
     return value.split(",")
+
+
+def pick_first(
+    config: dict, candidates: List[str], allow_empty: bool = False, allow_none: bool = False, allow_fail: bool = False
+):
+    """Find first valid entry in config for given keys.
+
+    Arguments
+    ---------
+    config : dict
+        The configuration data.
+    candidates : List[str]
+        The candidate keys.
+    allow_empty : bool, optional
+        Do not fail an return None if list is empty.
+    allow_none : bool, optional
+        Allow config entries with value None.
+    allow_fail : bool, optional
+        Do not fail if no valiud entry is found.
+
+    Returns
+    -------
+    ret : any
+        The found value in the config dict or None.
+
+    """
+    if len(candidates) == 0:
+        assert allow_empty, "pick_first: got empty list"
+        return None
+    ret = None
+    found = False
+    for candidate in candidates:
+        if candidate in config.keys():
+            val = config[candidate]
+            if val is not None or allow_none:
+                ret = val
+                found = True
+                break
+    if not found:
+        assert allow_fail, f"pick_first: could not find valid candidate in config ({candidates})"
+    return ret

{mlonmcu-0.7.dev0 → mlonmcu-0.8.0}/mlonmcu/context/context.py

@@ -22,7 +22,7 @@ import sys
 import os
 import shutil
 import tempfile
-from typing import List, Union
+from typing import List, Union, Optional
 from pathlib import Path
 import filelock
 
@@ -190,7 +190,7 @@ def load_recent_sessions(env: Environment, count: int = None) -> List[Session]:
             run.archived = True
             run.dir = run_directory
             runs.append(run)
-        session = Session(idx=sid, archived=True, dir=session_directory)
+        session = Session(idx=sid, archived=True, dest=session_directory)
         session.runs = runs
         session.dir = session_directory
         sessions.append(session)
@@ -308,7 +308,7 @@ class MlonMcuContext:
         self.cache = TaskCache()
         self.export_paths = set()
 
-    def create_session(self, label="", config=None):
+    def create_session(self, label="", config=None, dest: Optional[Union[str, Path]] = None):
         try:
             lock = self.latest_session_link_lock.acquire(timeout=10)
         except filelock.Timeout as err:
@@ -323,7 +323,7 @@ class MlonMcuContext:
                 sessions_directory = temp_directory / "sessions"
                 sessions_directory.mkdir(exist_ok=True, parents=True)
                 session_dir = sessions_directory / str(idx)
-                session = Session(idx=idx, label=label, dir=session_dir, config=config)
+                session = Session(idx=idx, label=label, dest=dest if dest is not None else session_dir, config=config)
                 self.sessions.append(session)
                 self.session_idx = idx
                 # TODO: move this to a helper function
@@ -350,6 +350,10 @@ class MlonMcuContext:
 
     def load_extensions(self):
         """If available load the extensions.py scripts in the plugin directories"""
+        if self.environment:
+            allow_extensions = self.environment.vars.get("allow_extensions", False)
+            if not allow_extensions:
+                return
 
         # TODO: check vars.enable_extensions before!
         def _load(plugins_dir, hint="Unknown"):
@@ -357,7 +361,7 @@ class MlonMcuContext:
                 extensions_file = plugins_dir / "extensions.py"
                 if extensions_file.is_file():
                     logger.info(f"Loading extensions.py ({hint})")
-                    process_extensions(extensions_file)
+                    process_extensions(extensions_file, desc=hint.lower())
 
         # global (user)
         plugins_dir = Path(get_plugins_dir())
@@ -370,7 +374,7 @@ class MlonMcuContext:
                     plugins_dir = self.environment.paths["plugins"].path
                     _load(plugins_dir, hint="Environment")
 
-    def get_session(self, label="", resume=False, config=None) -> Session:
+    def get_session(self, label="", resume=False, config=None, dest=None) -> Session:
         """Get an active session if available, else create a new one.
 
         Returns
@@ -385,8 +389,9 @@ class MlonMcuContext:
             raise NotImplementedError
 
         if self.session_idx < 0 or not self.sessions[-1].active:
-            self.create_session(label=label, config=config)
-        return self.sessions[-1]
+            self.create_session(label=label, config=config, dest=dest)
+        ret = self.sessions[-1]
+        return ret
 
     def __enter__(self):
         logger.debug("Enter MlonMcuContext")

{mlonmcu-0.7.dev0 → mlonmcu-0.8.0}/mlonmcu/feature/feature.py

@@ -16,7 +16,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-""" MLonMCU Features API"""
+"""MLonMCU Features API"""
 
 from abc import ABC
 

{mlonmcu-0.7.dev0 → mlonmcu-0.8.0}/mlonmcu/feature/features.py

@@ -155,7 +155,9 @@ class Muriscvnn(SetupFeature, FrameworkFeature, PlatformFeature):
     DEFAULTS = {
         **FeatureBase.DEFAULTS,
         "use_vext": "AUTO",
+        "use_portable": "AUTO",
         "use_pext": "AUTO",
+        "use_corev": False,
     }
 
     REQUIRED = {"muriscvnn.src_dir"}
@@ -175,6 +177,14 @@ class Muriscvnn(SetupFeature, FrameworkFeature, PlatformFeature):
         value = str2bool(value)
         return "ON" if value else "OFF"
 
+    @property
+    def use_portable(self):
+        value = self.config["use_portable"]
+        if value == "AUTO" or value is None:
+            return value
+        value = str2bool(value)
+        return "ON" if value else "OFF"
+
     @property
     def use_pext(self):
         value = self.config["use_pext"]
@@ -183,6 +193,12 @@ class Muriscvnn(SetupFeature, FrameworkFeature, PlatformFeature):
         value = str2bool(value)
         return "ON" if value else "OFF"
 
+    @property
+    def use_corev(self):
+        value = self.config["use_corev"]
+        value = str2bool(value)
+        return "ON" if value else "OFF"
+
     def add_framework_config(self, framework, config):
         assert framework == "tflm", f"Unsupported feature '{self.name}' for framework '{framework}'"
         if f"{framework}.optimized_kernel" in config and config[f"{framework}.optimized_kernel"] not in [
@@ -198,8 +214,10 @@ class Muriscvnn(SetupFeature, FrameworkFeature, PlatformFeature):
         return {
             "MURISCVNN": self.enabled,
             "MURISCVNN_DIR": self.muriscvnn_dir,
+            "MURISCVNN_PORTABLE": self.use_portable,
             "MURISCVNN_VEXT": self.use_vext,
             "MURISCVNN_PEXT": self.use_pext,
+            "MURISCVNN_COREV": self.use_corev,
         }
 
     def get_required_cache_flags(self):
@@ -339,6 +357,7 @@ class MuriscvnnByoc(SetupFeature, BackendFeature, PlatformFeature):
         "mattr": None,  # for +nodsp, +nomve
         "debug_last_error": False,
         "use_vext": "AUTO",
+        "use_portable": "AUTO",
         "use_pext": "AUTO",
     }
 
@@ -371,6 +390,14 @@ class MuriscvnnByoc(SetupFeature, BackendFeature, PlatformFeature):
         value = str2bool(value)
         return "ON" if value else "OFF"
 
+    @property
+    def use_portable(self):
+        value = self.config["use_portable"]
+        if value == "AUTO" or value is None:
+            return value
+        value = str2bool(value)
+        return "ON" if value else "OFF"
+
     @property
     def use_pext(self):
         value = self.config["use_pext"]
@@ -409,6 +436,7 @@ class MuriscvnnByoc(SetupFeature, BackendFeature, PlatformFeature):
         return {
             "MURISCVNN": self.enabled,
             "MURISCVNN_DIR": self.muriscvnn_dir,
+            "MURISCVNN_PORTABLE": self.use_portable,
             "MURISCVNN_VEXT": self.use_vext,
             "MURISCVNN_PEXT": self.use_pext,
         }
@@ -531,7 +559,11 @@ class Pext(SetupFeature, TargetFeature, PlatformFeature):
             "muriscvnn.lib": ["pext"],
             "tflmc.exe": ["pext"],
             "riscv_gcc.install_dir": ["pext"],
+            "riscv_gcc32.install_dir": ["pext"],
+            "riscv_gcc64.install_dir": ["pext"],
             "riscv_gcc.name": ["pext"],
+            "riscv_gcc32.name": ["pext"],
+            "riscv_gcc64.name": ["pext"],
         }
 
 
@@ -639,7 +671,7 @@ class GdbServer(TargetFeature):
         return int(self.config["port"]) if self.config["port"] is not None else None
 
     def get_target_config(self, target):
-        assert target in ["host_x86", "etiss_pulpino", "etiss", "ovpsim", "corev_ovpsim"]
+        assert target in ["host_x86", "etiss_pulpino", "etiss", "etiss_rv32", "etiss_rv64", "ovpsim", "corev_ovpsim"]
         return filter_none(
             {
                 f"{target}.gdbserver_enable": self.enabled,
@@ -678,10 +710,11 @@ class Trace(TargetFeature):
     def __init__(self, features=None, config=None):
         super().__init__("trace", features=features, config=config)
 
+    # TODO: add callback to save artifact?
     # def add_target_config(self, target, config, directory=None):
     def add_target_config(self, target, config):
-        assert target in ["etiss_pulpino", "etiss", "ovpsim"]
-        if target in ["etiss_pulpino", "etiss"]:
+        assert target in ["etiss_pulpino", "etiss", "etiss_perf", "ovpsim"]
+        if target in ["etiss_pulpino", "etiss", "etiss_perf"]:
             config.update({f"{target}.trace_memory": self.enabled})
         elif target == "ovpsim":
             extra_args_new = config.get("extra_args", [])
@@ -694,6 +727,39 @@ class Trace(TargetFeature):
                 extra_args_new.append(trace_file)
             config.update({f"{target}.extra_args": extra_args_new})
 
+    def get_target_callbacks(self, target):
+        assert target in [
+            "etiss_pulpino",
+            "etiss",
+            "etiss_perf",
+            "ovpsim",
+            "corev_ovpsim",
+        ], f"Unsupported feature '{self.name}' for target '{target}'"
+        if self.enabled:
+            if self.to_file:
+
+                def mem_trace_callback(stdout, metrics, artifacts, directory=None):
+                    """Callback which parses collects the generated artifacts."""
+                    file_name = "trace.txt"
+                    if target in ["etiss_pulpino", "etiss", "etiss_perf"]:
+                        file_name = "dBusAccess.csv"
+                    assert directory is not None
+                    file_path = Path(directory) / file_name
+                    assert file_path.is_file()
+                    with open(file_path, "r") as f:
+                        content = f.read()
+                    mem_artifact = Artifact(
+                        f"{target}_mem.log",
+                        content=content,
+                        fmt=ArtifactFormat.TEXT,
+                        flags=(self.name, target),
+                    )
+                    artifacts.append(mem_artifact)
+                    return stdout
+
+                return None, mem_trace_callback
+        return None, None
+
 
 @register_feature("unpacked_api")
 class UnpackedApi(BackendFeature):  # TODO: should this be a feature or config only?
@@ -791,7 +857,7 @@ class Usmp(BackendFeature):
 
 @register_feature("fuse_ops")
 class FuseOps(BackendFeature):
-    """TODO"""
+    """Exposes relay.FuseOps.max_depth settings for TVM backends."""
 
     DEFAULTS = {
         **FeatureBase.DEFAULTS,
@@ -1093,7 +1159,7 @@ class AutoTVM(TVMTuneBase):
 
 @register_feature("autoscheduler", depends=["autotune"])
 class AutoScheduler(TVMTuneBase):
-    """TODO"""
+    """TVM AutoScheduler Tuning Feature."""
 
     # TODO: metascheduler
     # TODO: graphtuner
@@ -1131,7 +1197,7 @@ class AutoScheduler(TVMTuneBase):
 
 @register_feature("metascheduler", depends=["autotune"])
 class MetaScheduler(TVMTuneBase):
-    """TODO"""
+    """TVM MetaScheduler Tuning Feature."""
 
     DEFAULTS = {
         **TVMTuneBase.DEFAULTS,
@@ -1386,7 +1452,16 @@ class LogInstructions(TargetFeature):
 
     # def add_target_config(self, target, config, directory=None):
     def add_target_config(self, target, config):
-        assert target in ["spike", "etiss_pulpino", "etiss", "ovpsim", "corev_ovpsim", "gvsoc_pulp"]
+        assert target in [
+            "spike",
+            "etiss_pulpino",
+            "etiss",
+            "etiss_perf",
+            "ovpsim",
+            "corev_ovpsim",
+            "gvsoc_pulp",
+            "vicuna2",
+        ]
         if not self.enabled:
             return
         if target == "spike":
@@ -1398,7 +1473,7 @@ class LogInstructions(TargetFeature):
                 log_file = directory / "instrs.txt"
                 extra_args_new.append(f"--log={log_file}")
             config.update({f"{target}.extra_args": extra_args_new})
-        elif target in ["etiss_pulpino", "etiss"]:
+        elif target in ["etiss_pulpino", "etiss", "etiss_perf"]:
             plugins_new = config.get("plugins", [])
             plugins_new.append("PrintInstruction")
             config.update({f"{target}.plugins": plugins_new})
@@ -1427,15 +1502,19 @@ class LogInstructions(TargetFeature):
             else:
                 extra_args_new.append("--trace=insn")
             config.update({f"{target}.extra_args": extra_args_new})
+        elif target == "vicuna2":
+            config.update({f"{target}.log_instrs": True})
 
     def get_target_callbacks(self, target):
         assert target in [
             "spike",
             "etiss_pulpino",
             "etiss",
+            "etiss_perf",
             "ovpsim",
             "corev_ovpsim",
             "gvsoc_pulp",
+            "vicuna2",
         ], f"Unsupported feature '{self.name}' for target '{target}'"
         if self.enabled:
             if not target == "gvsoc_pulp":
@@ -1444,14 +1523,14 @@ class LogInstructions(TargetFeature):
                     """Callback which parses the targets output and updates the generated metrics and artifacts."""
                     new_lines = []
                     if self.to_file:
-                        if target in ["etiss_pulpino", "etiss"]:
+                        if target in ["etiss_pulpino", "etiss", "etiss_perf"]:
                             if self.etiss_experimental_print_to_file:
                                 log_file = Path(directory) / "instr_trace.csv"
                             else:
                                 # TODO: update stdout and remove log_instrs lines
                                 instrs = []
                                 for line in stdout.split("\n"):
-                                    if target in ["etiss_pulpino", "etiss"]:
+                                    if target in ["etiss_pulpino", "etiss", "etiss_perf"]:
                                         expr = re.compile(r"0x[a-fA-F0-9]+: .* \[.*\]")
                                     match = expr.match(line)
                                     if match is not None:
@@ -1460,6 +1539,10 @@ class LogInstructions(TargetFeature):
                                         new_lines.append(line)
                                 content = "\n".join(instrs)
                                 stdout = "\n".join(new_lines)
+                        elif target == "vicuna2":
+                            log_file = Path(directory) / "log_instrs.csv"
+                            with open(log_file, "r") as f:
+                                content = f.read()
                         else:
                             assert target in ["spike", "ovpsim", "corev_ovpsim"]
                             log_file = Path(directory) / "instrs.txt"
@@ -2205,7 +2288,7 @@ class CV32HpmCounter(HpmCounter):  # TODO: SetupFeature?
 
 @register_feature("vanilla_accelerator")
 class VanillaAccelerator(TargetFeature):
-    """TODO"""
+    """BYOC Vanilla Accelerator Feature for TVM."""
 
     DEFAULTS = {
         **FeatureBase.DEFAULTS,
@@ -2447,3 +2530,123 @@ class ValidateNew(RunFeature):
     #     # validate_outputs_postprocess = ValidateOutputsPostprocess(features=[], config=config)
     #     # return [validate_outputs_postprocess]
     #     return ["validate_outputs"]
+
+
+@register_feature("perf_sim")
+class PerfSim(TargetFeature):
+    """ETISS Performance Estimation/Simulation Feature."""
+
+    DEFAULTS = {
+        **FeatureBase.DEFAULTS,
+        "core": "cv32e40p",
+        "estimator": True,
+        "trace_asm": False,  # TODO: move to owen feature, save files!
+        "trace_instr": False,  # TODO: move to owen feature, save files!
+        "to_file": False,
+    }
+
+    def __init__(self, features=None, config=None):
+        super().__init__("perf_sim", features=features, config=config)
+
+    @property
+    def core(self):
+        value = self.config["core"]
+        return value
+
+    @property
+    def estimator(self):
+        value = self.config["estimator"]
+        return str2bool(value)
+
+    @property
+    def trace_asm(self):
+        value = self.config["trace_asm"]
+        return str2bool(value)
+
+    @property
+    def trace_instr(self):
+        value = self.config["trace_instr"]
+        return str2bool(value)
+
+    def to_file(self):
+        value = self.config["to_file"]
+        return str2bool(value, allow_none=True)
+
+    def add_target_config(self, target, config):
+        assert target in ["etiss_perf"]
+        if not self.enabled:
+            return
+        plugins_new = config.get(f"{target}.plugins", [])
+        extra_plugin_config = config.get(f"{target}.extra_plugin_config", {})
+        if self.estimator:
+            plugins_new.append("PerformanceEstimatorPlugin")
+            if self.core is not None:
+                assert self.name not in extra_plugin_config
+                extra_plugin_config["perfEst"] = {}
+                extra_plugin_config["perfEst"]["uArch"] = self.core.upper()
+                if self.to_file:
+                    extra_plugin_config["perfEst"]["print"] = 1
+                extra_plugin_config["perfEst"]["printDir"] = "."
+        if self.trace_asm or self.trace_instr:
+            if self.to_file:
+                assert not (self.trace_asm and self.trace_instr)
+                assert not self.trace_instr or self.core.upper() == "CVA6"
+                plugins_new.append("TracePrinterPlugin")
+                trace = "InstructionTrace_RV64" if self.trace_instr else "AssemblyTrace"
+                # trace = "InstructionTrace_RV64" if self.trace_instr else "CV32E40P"
+                extra_plugin_config["tracePrinter"] = {}
+                extra_plugin_config["tracePrinter"]["trace"] = trace
+                extra_plugin_config["tracePrinter"]["stream.toFile"] = 1
+                # extra_plugin_config["tracePrinter"]["outDir"] = "/tmp/perf_out/"  # TODO: do not hardcode
+                extra_plugin_config["tracePrinter"]["stream.outDir"] = "."
+                extra_plugin_config["tracePrinter"]["stream.fileName"] = (
+                    "instr_trace" if self.trace_instr else "asm_trace"
+                )
+                extra_plugin_config["tracePrinter"]["stream.postfix"] = ".csv"
+                extra_plugin_config["tracePrinter"]["rotateSize"] = 0x100000
+        config.update({f"{target}.plugins": plugins_new})
+        config.update({f"{target}.extra_plugin_config": extra_plugin_config})
+
+    def get_target_callbacks(self, target):
+        assert target in ["etiss_perf"]
+        if self.enabled:
+
+            def metrics_callback(stdout, metrics, artifacts, directory=None):
+                """Callback for extracting perf metrics from stdout"""
+                assert len(metrics) == 1
+                if self.estimator:
+                    instrs_match = re.compile(r" >> Number of instructions: (\d+)").findall(stdout)
+                    assert instrs_match is not None
+                    assert len(instrs_match) == 1
+                    cycles_match = re.compile(r" >> Estimated number of processor cycles: (\d+)").findall(stdout)
+                    assert cycles_match is not None
+                    assert len(cycles_match) == 1
+                    cpi_match = re.compile(
+                        r" >> Estimated average number of processor cycles per instruction: (\d+?\.\d+)"
+                    ).findall(stdout)
+                    assert cpi_match is not None
+                    assert len(cpi_match) == 1
+                    metrics[0].add("PerfSim Instructions", int(instrs_match[0]))
+                    metrics[0].add("PerfSim Cycles", int(cycles_match[0]))
+                    metrics[0].add("PerfSim CPI", float(cpi_match[0]))
+                if self.to_file:
+                    assert directory is not None
+                    assert self.core is not None
+                    csv_files = list(Path(directory).glob(f"{self.core.upper()}_*.csv"))
+                    if self.trace_asm:
+                        csv_files_ = list(Path(directory).glob("asm_trace_*.csv"))
+                        csv_files += csv_files_
+                    for file in csv_files:
+                        with open(file, "r") as f:
+                            content = f.read()
+                        artifact = Artifact(
+                            file.name,
+                            content=content,
+                            fmt=ArtifactFormat.TEXT,
+                            flags=(self.name, target),
+                        )
+                        artifacts.append(artifact)
+                return stdout
+
+            return None, metrics_callback
+        return None, None

{mlonmcu-0.7.dev0 → mlonmcu-0.8.0}/mlonmcu/flow/backend.py

@@ -72,7 +72,9 @@ class Backend(ABC):
         return features
 
     @abstractmethod
-    def load_model(self, model, input_shapes=None, output_shapes=None, input_types=None, output_types=None):
+    def load_model(
+        self, model, input_shapes=None, output_shapes=None, input_types=None, output_types=None, params_path=None
+    ):
         pass
 
     @abstractmethod

{mlonmcu-0.7.dev0 → mlonmcu-0.8.0}/mlonmcu/flow/tflm/backend/backend.py

@@ -42,7 +42,10 @@ class TFLMBackend(Backend):
         assert isinstance(cls.name, str)
         cls.registry[cls.name] = cls
 
-    def load_model(self, model, input_shapes=None, output_shapes=None, input_types=None, output_types=None):
+    def load_model(
+        self, model, input_shapes=None, output_shapes=None, input_types=None, output_types=None, params_path=None
+    ):
+        assert params_path is None
         self.model = model
         ext = os.path.splitext(model)[1][1:]
         fmt = ModelFormats.from_extension(ext)