tum-ei-eda / mlonmcu

Tool for the deployment and analysis of TinyML applications on TFLM and MicroTVM backends
Apache License 2.0
29 stars 13 forks source link

Reduce RAM consumption during MLonMCU Flow #155

Open PhilippvK opened 6 months ago

PhilippvK commented 6 months ago

When running large benchmarks session with intruction-level tracing I am sometimes seeing peek RAM usage of ~100GB on a workstation equipped with 128GB. The situation can be described es follows:

The first and second plot in this figure should visualize this behavior for an example session (thread_pool_ps peaking at ~40GB RAM for the approach using ThreadPoolExecutor and runs_per_stage=1, see #153 for more context):

mlonmcu_ram_cpu_disk

Here are some thoughts about potential solutions for the mentioned problems: