change to use to_edge_transform_and_lower

Author: Gasoonjia

backends/aoti/runtime/aoti_backend.cpp

@@ -352,7 +352,6 @@ class AOTIBackend final : public ::executorch::runtime::BackendInterface {
   }
 
   void destroy(DelegateHandle* handle_) const override {
-    ET_LOG(Debug, "AOTIBackend handle %p destroy", handle_);
     AOTIDelegateHandle* handle = (AOTIDelegateHandle*)handle_;
 
     // Delete the container BEFORE closing the shared library
@@ -364,11 +363,6 @@ class AOTIBackend final : public ::executorch::runtime::BackendInterface {
             Error,
             "AOTInductorModelContainerDelete failed with error code %d",
             delete_result);
-      } else {
-        ET_LOG(
-            Debug,
-            "AOTIBackend container_handle %p deleted",
-            handle->container_handle);
       }
     }
 
@@ -380,6 +374,7 @@ class AOTIBackend final : public ::executorch::runtime::BackendInterface {
     free(handle);
     cleanup_memory();
     cleanup_tensor_metadata();
+    ET_LOG(Debug, "AOTIBackend handle %p destroy", handle_);
   }
 };
 

export_aoti.py

@@ -21,7 +21,7 @@
 
 import torch
 from executorch.backends.aoti.aoti_partitioner import AotiPartitioner
-from executorch.exir import to_edge
+from executorch.exir import to_edge_transform_and_lower
 from torch import nn
 from torch.export import export
 from torchvision import models
@@ -135,13 +135,17 @@ def export_model(model, example_inputs, output_filename="aoti_model.pte"):
     aten_dialect = export(model, example_inputs)
 
     # 2. to_edge: Make optimizations for Edge devices
-    print("Step 2: Converting to Edge program...")
-    edge_program = to_edge(aten_dialect)
-    print(edge_program.exported_program().graph.print_tabular())
+    # print("Step 2: Converting to Edge program...")
+    # edge_program = to_edge(aten_dialect)
+    # print(edge_program.exported_program().graph.print_tabular())
 
-    print("Step 3: Converting to backend...")
-    edge_program = edge_program.to_backend(AotiPartitioner([]))
-    print("To backend done.")
+    # print("Step 3: Converting to backend...")
+    # edge_program = edge_program.to_backend(AotiPartitioner([]))
+    # print("To backend done.")
+
+    edge_program = to_edge_transform_and_lower(
+        aten_dialect, partitioner=[AotiPartitioner([])]
+    )
 
     # 3. to_executorch: Convert the graph to an ExecuTorch program
     print("Step 4: Converting to ExecuTorch program...")