ExtensionLLM: Make it possible to pass memory mode for the Ethos-U

Signed-off-by: George Gekov <george.gekov@arm.com>
Change-Id: I9be36752f199c32a26a69fb8b4a99edde5a6f2a2

Author: gggekov

examples/models/llama/export_llama_lib.py

@@ -1035,6 +1035,8 @@ def _to_edge_and_lower_llama_arm(
         partitioners.append(
             get_ethosu_partitioner(
                 llm_config.backend.ethosu.target,
+                llm_config.backend.ethosu.system_config,
+                llm_config.backend.ethosu.memory_mode,
             )
         )
         modelname = f"ethosu_{modelname}"

extension/llm/export/partitioner_lib.py

@@ -248,11 +248,19 @@ def get_tosa_partitioner(version: str):
     return TOSAPartitioner(compile_spec)
 
 
-def get_ethosu_partitioner(target: str):
+def get_ethosu_partitioner(
+    target: str,
+    system_config: Optional[str] = None,
+    memory_mode: Optional[str] = None,
+):
     from executorch.backends.arm.ethosu.compile_spec import EthosUCompileSpec
     from executorch.backends.arm.ethosu.partitioner import EthosUPartitioner
 
-    compile_spec = EthosUCompileSpec(target)
+    compile_spec = EthosUCompileSpec(
+        target,
+        system_config=None if system_config == "default" else system_config,
+        memory_mode=None if memory_mode == "default" else memory_mode,
+    )
 
     return EthosUPartitioner(compile_spec)