remove convert in compiler_funcs

Differential Revision: D81065417

Pull Request resolved: #13698

Author: ethansfng

backends/cadence/aot/compiler.py

@@ -13,7 +13,6 @@
 import executorch.backends.cadence.aot.ops_registrations  # noqa
 import torch
 from executorch.backends.cadence.aot.compiler_funcs import (
-    convert as convert_fn,
     prepare as prepare_fn,
     trace as trace_fn,
 )
@@ -42,6 +41,7 @@
 from executorch.exir.program._program import to_edge
 
 from torch.export.exported_program import ExportedProgram
+from torchao.quantization.pt2e.quantize_pt2e import convert_pt2e
 
 from .passes import apply_exir_ops_passes, apply_torch_ops_passes
 
@@ -139,7 +139,7 @@ def convert_pt2(
     Returns a GraphModule with the converted model.
     """
 
-    converted_model = convert_fn(graph_module)
+    converted_model = convert_pt2e(graph_module)
 
     if dump_graphs:
         logging.info("Graph after convert:")

backends/cadence/aot/compiler_funcs.py

@@ -11,11 +11,7 @@
 
 import torch
 from torch._inductor.decomposition import remove_decompositions
-from torchao.quantization.pt2e.quantize_pt2e import (
-    convert_pt2e,
-    prepare_pt2e,
-    prepare_qat_pt2e,
-)
+from torchao.quantization.pt2e.quantize_pt2e import prepare_pt2e, prepare_qat_pt2e
 from torchao.quantization.pt2e.quantizer import Quantizer
 
 
@@ -56,8 +52,3 @@ def prepare(
         prepared_model = prepare_pt2e(traced_model, quantizer)
 
     return prepared_model
-
-
-def convert(prepared_model: torch.fx.GraphModule) -> torch.fx.GraphModule:
-    converted_model = convert_pt2e(prepared_model)
-    return converted_model