NXP backend: Unify pipeines in aot_neutron_compile.py and executorch_pipeline.py

Author: StrycekSimon

backends/nxp/tests/executorch_pipeline.py

@@ -117,6 +117,11 @@ def to_quantized_edge_program(
 
     edge_program_manager = NeutronEdgePassManager()(edge_program_manager)
 
+    if remove_quant_io_ops:
+        edge_program_manager = edge_program_manager.transform(
+            [RemoveIOQuantOpsPass(edge_program_manager=edge_program_manager)]
+        )
+
     compile_spec = generate_neutron_compile_spec(
         target,
         operators_not_to_delegate=operators_not_to_delegate,
@@ -125,11 +130,6 @@ def to_quantized_edge_program(
     partitioner = NeutronPartitioner(compile_spec, custom_delegation_options)
     edge_program_manager = edge_program_manager.to_backend(partitioner)
 
-    if remove_quant_io_ops:
-        edge_program_manager = edge_program_manager.transform(
-            [RemoveIOQuantOpsPass(edge_program_manager=edge_program_manager)]
-        )
-
     return edge_program_manager
 
 

examples/nxp/aot_neutron_compile.py

@@ -272,15 +272,15 @@ def _get_batch_size(data):
     # 5. Delegate to Neutron
     if args.delegate:
         logging.info("Executing Neutron Partitioner and Delegate")
-        edge_program_manager = edge_program_manager.to_backend(
-            NeutronPartitioner(
-                generate_neutron_compile_spec(
-                    args.target,
-                    args.neutron_converter_flavor,
-                    operators_not_to_delegate=args.operators_not_to_delegate,
-                )
-            )
+
+        compile_spec = generate_neutron_compile_spec(
+            args.target,
+            operators_not_to_delegate=args.operators_not_to_delegate,
+            neutron_converter_flavor=args.neutron_converter_flavor,
         )
+        partitioner = NeutronPartitioner(compile_spec)
+
+        edge_program_manager = edge_program_manager.to_backend(partitioner)
         logging.debug(
             f"Lowered graph:\n{edge_program_manager.exported_program().graph}"
         )