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

Author: StrycekSimon

backends/nxp/tests/executorch_pipeline.py

@@ -119,14 +119,15 @@ def to_quantized_edge_program(
         operators_not_to_delegate=operators_not_to_delegate,
         neutron_converter_flavor=neutron_converter_flavor,
     )
-    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)]
         )
 
+    partitioner = NeutronPartitioner(compile_spec, custom_delegation_options)
+    edge_program_manager = edge_program_manager.to_backend(partitioner)
+
     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}"
         )