mlonmcu 0.6.0__tar.gz → 0.6.dev0__tar.gz

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

@@ -1,13 +1,12 @@
 Metadata-Version: 2.1
 Name: mlonmcu
-Version: 0.6.0
+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
 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
@@ -21,7 +20,6 @@ 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
@@ -37,6 +35,7 @@ Provides-Extra: tvm-autotuning
 Provides-Extra: visualize
 Provides-Extra: visualize_tflite
 Provides-Extra: zephyr
+Provides-Extra: all-prod
 License-File: LICENSE
 License-File: NOTICE
 
@@ -261,5 +260,3 @@ the projects [Scale4Edge](https://www.edacentrum.de/scale4edge/) (grant number 1
 ## 0.1.0 (2021-11-12)
 
 * First closed-source release.
-
-

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

@@ -48,7 +48,7 @@ class FeatureBase(ABC):
     @property
     def enabled(self):
         value = self.config["enabled"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     def remove_config_prefix(self, config):  # TODO: move to different place
         def helper(key):

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

@@ -127,7 +127,7 @@ class Validate(FrontendFeature, PlatformFeature):
     @property
     def allow_missing(self):
         value = self.config["allow_missing"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def fail_on_error(self):
@@ -172,7 +172,8 @@ class Muriscvnn(SetupFeature, FrameworkFeature, PlatformFeature):
         value = self.config["use_vext"]
         if value == "AUTO" or value is None:
             return value
-        value = str2bool(value)
+        if not isinstance(value, (bool, int)):
+            value = str2bool(value)
         return "ON" if value else "OFF"
 
     @property
@@ -180,7 +181,8 @@ class Muriscvnn(SetupFeature, FrameworkFeature, PlatformFeature):
         value = self.config["use_pext"]
         if value == "AUTO" or value is None:
             return value
-        value = str2bool(value)
+        if not isinstance(value, (bool, int)):
+            value = str2bool(value)
         return "ON" if value else "OFF"
 
     def add_framework_config(self, framework, config):
@@ -368,7 +370,8 @@ class MuriscvnnByoc(SetupFeature, BackendFeature, PlatformFeature):
         value = self.config["use_vext"]
         if value == "AUTO" or value is None:
             return value
-        value = str2bool(value)
+        if not isinstance(value, (bool, int)):
+            value = str2bool(value)
         return "ON" if value else "OFF"
 
     @property
@@ -376,7 +379,8 @@ class MuriscvnnByoc(SetupFeature, BackendFeature, PlatformFeature):
         value = self.config["use_pext"]
         if value == "AUTO" or value is None:
             return value
-        value = str2bool(value)
+        if not isinstance(value, (bool, int)):
+            value = str2bool(value)
         return "ON" if value else "OFF"
 
     def add_backend_config(self, backend, config):
@@ -559,22 +563,22 @@ class Bext(SetupFeature, TargetFeature, PlatformFeature):
     @property
     def zba(self):
         value = self.config["zba"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def zbb(self):
         value = self.config["zbb"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def zbc(self):
         value = self.config["zbc"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def zbs(self):
         value = self.config["zbs"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     def get_target_config(self, target):
         return filter_none(
@@ -632,7 +636,7 @@ class GdbServer(TargetFeature):
     @property
     def attach(self):
         value = self.config["attach"]
-        return str2bool(value, allow_none=True)
+        return str2bool(value, allow_none=True) if not isinstance(value, (bool, int)) else value
 
     @property
     def port(self):
@@ -673,7 +677,7 @@ class Trace(TargetFeature):
     @property
     def to_file(self):
         value = self.config["to_file"]
-        return str2bool(value, allow_none=True)
+        return str2bool(value, allow_none=True) if not isinstance(value, (bool, int)) else value
 
     def __init__(self, features=None, config=None):
         super().__init__("trace", features=features, config=config)
@@ -761,7 +765,7 @@ class Usmp(BackendFeature):
     @property
     def use_workspace_io(self):
         value = self.config["use_workspace_io"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     def add_backend_config(self, backend, config):
         assert backend in ["tvmaot"], f"Unsupported feature '{self.name}' for backend '{backend}'"
@@ -1282,7 +1286,7 @@ class CacheSim(TargetFeature):
     @property
     def ic_enable(self):
         value = self.config["ic_enable"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def ic_config(self):
@@ -1291,7 +1295,7 @@ class CacheSim(TargetFeature):
     @property
     def dc_enable(self):
         value = self.config["dc_enable"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def dc_config(self):
@@ -1300,7 +1304,7 @@ class CacheSim(TargetFeature):
     @property
     def l2_enable(self):
         value = self.config["l2_enable"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def l2_config(self):
@@ -1309,12 +1313,12 @@ class CacheSim(TargetFeature):
     @property
     def log_misses(self):
         value = self.config["log_misses"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def detailed(self):
         value = self.config["detailed"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     # def add_target_config(self, target, config, directory=None):
     def add_target_config(self, target, config):
@@ -1375,7 +1379,7 @@ class LogInstructions(TargetFeature):
     @property
     def to_file(self):
         value = self.config["to_file"]
-        return str2bool(value, allow_none=True)
+        return str2bool(value, allow_none=True) if not isinstance(value, (bool, int)) else value
 
     # def add_target_config(self, target, config, directory=None):
     def add_target_config(self, target, config):
@@ -1523,12 +1527,12 @@ class TargetOptimized(RunFeature):
     @property
     def layouts(self):
         value = self.config["layouts"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def schedules(self):
         value = self.config["schedules"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     def __init__(self, features=None, config=None):
         super().__init__("target_optimized", features=features, config=config)
@@ -1567,7 +1571,7 @@ class AutoVectorize(PlatformFeature):
     @property
     def verbose(self):
         value = self.config["verbose"]
-        # return str2bool(value)
+        # return str2bool(value) if not isinstance(value, (bool, int)) else value
         if value is None or not value:
             return "OFF"
         assert isinstance(value, str)
@@ -1578,12 +1582,12 @@ class AutoVectorize(PlatformFeature):
     @property
     def loop(self):
         value = self.config["loop"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def slp(self):
         value = self.config["slp"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def force_vector_width(self):
@@ -1614,7 +1618,7 @@ class AutoVectorize(PlatformFeature):
     @property
     def custom_unroll(self):
         value = self.config["custom_unroll"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     def get_platform_defs(self, platform):
         return {
@@ -1656,7 +1660,7 @@ class Benchmark(PlatformFeature, TargetFeature):
     @property
     def total(self):
         value = self.config["total"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def aggregate(self):
@@ -1844,37 +1848,37 @@ class XCoreV(TargetFeature, PlatformFeature, SetupFeature):
     @property
     def mac(self):
         value = self.config["mac"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def mem(self):
         value = self.config["mem"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def bi(self):
         value = self.config["bi"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def alu(self):
         value = self.config["alu"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def bitmanip(self):
         value = self.config["bitmanip"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def simd(self):
         value = self.config["simd"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def hwlp(self):
         value = self.config["hwlp"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     # def add_target_config(self, target, config, directory=None):
     def add_target_config(self, target, config):
@@ -1936,8 +1940,8 @@ class Xpulp(TargetFeature, PlatformFeature, SetupFeature):
 
     # Default getter functions:
     @staticmethod
-    def generalized_str2bool(inp: Union[str, bool, int]) -> bool:
-        return str2bool(inp)
+    def generalized_str2bool(input: Union[str, bool, int]) -> bool:
+        return str2bool(input) if isinstance(input, str) else input
 
     def getter_bool(self, key_name: Union[str, bool, int]) -> bool:
         return self.generalized_str2bool(self.config[key_name])

{mlonmcu-0.6.0 → mlonmcu-0.6.dev0}/mlonmcu/flow/tflm/backend/tflmc.py

@@ -50,7 +50,7 @@ class TFLMCBackend(TFLMBackend):
     @property
     def print_outputs(self):
         value = self.config["print_outputs"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     def __init__(self, features=None, config=None):
         super().__init__(features=features, config=config)

{mlonmcu-0.6.0 → mlonmcu-0.6.dev0}/mlonmcu/flow/tflm/backend/tflmi.py

@@ -20,7 +20,7 @@ import sys
 from typing import Tuple
 
 from .backend import TFLMBackend
-from mlonmcu.config import str2bool, str2list, str2dict
+from mlonmcu.config import str2bool, str2list
 from mlonmcu.flow.backend import main
 from mlonmcu.artifact import Artifact, ArtifactFormat
 
@@ -435,12 +435,12 @@ class TFLMIBackend(TFLMBackend):
     @property
     def legacy(self):
         value = self.config["legacy"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def debug_arena(self):
         value = self.config["debug_arena"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def arena_size(self):
@@ -449,17 +449,17 @@ class TFLMIBackend(TFLMBackend):
     @property
     def ops(self):
         value = self.config["ops"]
-        return str2list(value)
+        return str2list(value) if isinstance(value, str) else value
 
     @property
     def custom_ops(self):
         value = self.config["custom_ops"]
-        return str2list(value)
+        return str2list(value) if isinstance(value, str) else value
 
     @property
     def registrations(self):
         value = self.config["registrations"]
-        return str2dict(value)
+        return str2list(value) if isinstance(value, str) else value
 
     @property
     def ops_resolver(self):
@@ -468,7 +468,7 @@ class TFLMIBackend(TFLMBackend):
     @property
     def reporter(self):
         value = self.config["reporter"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     def generate(self) -> Tuple[dict, dict]:
         artifacts = []

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

@@ -252,7 +252,7 @@ class TVMBackend(Backend):
     @property
     def use_tuning_results(self):
         value = self.config["use_tuning_results"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def tvmc_extra_args(self):
@@ -265,7 +265,7 @@ class TVMBackend(Backend):
     @property
     def disabled_passes(self):
         value = self.config["disabled_passes"]
-        return str2list(value)
+        return str2list(value) if isinstance(value, str) else value
 
     @property
     def tvm_pythonpath(self):
@@ -286,17 +286,17 @@ class TVMBackend(Backend):
     @property
     def print_outputs(self):
         value = self.config["print_outputs"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def use_tlcpack(self):
         value = self.config["tvm.use_tlcpack"]
-        return str2bool(value, allow_none=True)
+        return str2bool(value, allow_none=True) if not isinstance(value, (bool, int)) else value
 
     @property
     def custom_unroll(self):
         value = self.config["custom_unroll"]
-        return str2bool(value, allow_none=True)
+        return str2bool(value, allow_none=True) if not isinstance(value, (bool, int)) else value
 
     @property
     def dump(self):
@@ -318,7 +318,7 @@ class TVMBackend(Backend):
     @property
     def refresh_model_info(self):
         value = self.config["refresh_model_info"]
-        return str2bool(value, allow_none=True)
+        return str2bool(value, allow_none=True) if not isinstance(value, (bool, int)) else value
 
     @property
     def generate_wrapper(self):

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

@@ -67,12 +67,12 @@ class TVMAOTBackend(TVMBackend):
     @property
     def unpacked_api(self):
         value = self.config["unpacked_api"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def debug_arena(self):
         value = self.config["debug_arena"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def alignment_bytes(self):

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

@@ -65,7 +65,7 @@ class TVMLLVMBackend(TVMBackend):
     @property
     def debug_arena(self):
         value = self.config["debug_arena"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     def get_tvmc_compile_args(self, out, dump=None):
         return super().get_tvmc_compile_args(out, dump=dump) + get_tvmrt_tvmc_args(

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

@@ -55,7 +55,7 @@ class TVMRTBackend(TVMBackend):
     @property
     def debug_arena(self):
         value = self.config["debug_arena"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     def get_tvmc_compile_args(self, out, dump=None):
         return super().get_tvmc_compile_args(out, dump=dump) + get_tvmrt_tvmc_args(system_lib=self.system_lib)

{mlonmcu-0.6.0 → mlonmcu-0.6.dev0}/mlonmcu/models/frontend.py

@@ -99,12 +99,12 @@ class Frontend(ABC):
     @property
     def use_inout_data(self):
         value = self.config["use_inout_data"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def gen_data(self):
         value = self.config["gen_data"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def gen_data_fill_mode(self):
@@ -129,7 +129,7 @@ class Frontend(ABC):
     @property
     def gen_ref_data(self):
         value = self.config["gen_ref_data"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def gen_ref_data_mode(self):
@@ -150,7 +150,7 @@ class Frontend(ABC):
     @property
     def gen_ref_labels(self):
         value = self.config["gen_ref_labels"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def gen_ref_labels_mode(self):
@@ -956,12 +956,12 @@ class TfLiteFrontend(SimpleFrontend):
     @property
     def visualize_enable(self):
         value = self.config["visualize_enable"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def split_layers(self):
         value = self.config["split_layers"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def visualize_script(self):
@@ -974,7 +974,7 @@ class TfLiteFrontend(SimpleFrontend):
     @property
     def analyze_enable(self):
         value = self.config["analyze_enable"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def analyze_script(self):
@@ -1238,7 +1238,7 @@ class RelayFrontend(SimpleFrontend):
     @property
     def visualize_graph(self):
         value = self.config["visualize_graph"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def relayviz_plotter(self):
@@ -1366,22 +1366,22 @@ class PackedFrontend(Frontend):  # Inherit from TFLiteFrontend? -> how to do con
     @property
     def ignore_existing(self):
         value = self.config["ignore_existing"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def fake_pack(self):
         value = self.config["fake_pack"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def use_packed(self):
         value = self.config["use_packed"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def check(self):
         value = self.config["check"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     def produce_artifacts(self, model):
         tflite_data = None

{mlonmcu-0.6.0 → mlonmcu-0.6.dev0}/mlonmcu/platform/espidf/espidf.py

@@ -92,18 +92,18 @@ class EspIdfPlatform(CompilePlatform, TargetPlatform):
     @property
     def use_idf_monitor(self):
         value = self.config["use_idf_monitor"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def wait_for_user(self):
         value = self.config["wait_for_user"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def flash_only(self):
         # TODO: get rid of this
         value = self.config["flash_only"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     def invoke_idf_exe(self, *args, **kwargs):
         env = os.environ.copy()

{mlonmcu-0.6.0 → mlonmcu-0.6.dev0}/mlonmcu/platform/microtvm/microtvm_corev_ovpsim_target.py

@@ -96,37 +96,37 @@ class CoreVOVPSimMicroTvmPlatformTarget(TemplateMicroTvmPlatformTarget):
     @property
     def enable_xcorevmac(self):
         value = self.config["enable_xcorevmac"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def enable_xcorevmem(self):
         value = self.config["enable_xcorevmem"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def enable_xcorevbi(self):
         value = self.config["enable_xcorevbi"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def enable_xcorevalu(self):
         value = self.config["enable_xcorevalu"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def enable_xcorevbitmanip(self):
         value = self.config["enable_xcorevbitmanip"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def enable_xcorevsimd(self):
         value = self.config["enable_xcorevsimd"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def enable_xcorevhwlp(self):
         value = self.config["enable_xcorevhwlp"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     def get_project_options(self):
         ret = super().get_project_options()

{mlonmcu-0.6.0 → mlonmcu-0.6.dev0}/mlonmcu/platform/microtvm/microtvm_etiss_target.py

@@ -104,37 +104,37 @@ class EtissMicroTvmPlatformTarget(TemplateMicroTvmPlatformTarget):
     @property
     def enable_xcorevmac(self):
         value = self.config["enable_xcorevmac"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def enable_xcorevmem(self):
         value = self.config["enable_xcorevmem"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def enable_xcorevbi(self):
         value = self.config["enable_xcorevbi"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def enable_xcorevalu(self):
         value = self.config["enable_xcorevalu"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def enable_xcorevbitmanip(self):
         value = self.config["enable_xcorevbitmanip"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def enable_xcorevsimd(self):
         value = self.config["enable_xcorevsimd"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     @property
     def enable_xcorevhwlp(self):
         value = self.config["enable_xcorevhwlp"]
-        return str2bool(value)
+        return str2bool(value) if not isinstance(value, (bool, int)) else value
 
     def get_project_options(self):
         ret = super().get_project_options()