NXP backend: Catching the converter failures.

Author: jirioc

backends/nxp/backend/neutron_converter_manager.py

@@ -3,11 +3,18 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 import importlib
+import logging
+import multiprocessing
 import pkgutil
 
 from executorch.backends.nxp.backend.ir.converter.node_converter import Target
 
 
+def convert_unsafe(neutron_converter, tflite_model, cctx, queue):
+    model_converted = neutron_converter.convertModel(list(tflite_model), cctx)
+    queue.put(model_converted)
+
+
 class NeutronConverterManager:
     """
     Manager for conversion of TFLite model in flatbuffers format into TFLite model that
@@ -52,6 +59,25 @@ def convert(
         cctx.targetOpts = neutron_converter.getNeutronTarget(target)
         # New switch since Neutron Converter SDK_25.06
         cctx.compilationOpts.minNumOpsPerGraph = 1
-        model_converted = neutron_converter.convertModel(list(tflite_model), cctx)
+        # Don't merge `Transpose` operators, to avoid creating unsupported permutations.
+        cctx.compilationOpts.excludeGraphPasses = "MergeTranspose"
+
+        logger = multiprocessing.log_to_stderr()
+        logger.setLevel(logging.WARNING)
+        queue = multiprocessing.Manager().Queue()
+
+        process = multiprocessing.Process(
+            target=convert_unsafe, args=(neutron_converter, tflite_model, cctx, queue)
+        )
+        process.start()
+        process.join()  # waits until the subprocess is complete
+
+        if queue.empty():  # signals the unsafe task did not run till the end
+            raise RuntimeError(
+                f"Neutron converter module terminated unexpectedly with exit code {process.exitcode}"
+            )
+
+        model_converted = queue.get()
 
+        process.close()
         return bytes(model_converted)