From 99dce34d885fdb44f96f27a81c6d1c875f3767e7 Mon Sep 17 00:00:00 2001
From: Ethan Ng <ethann@meta.com>
Date: Thu, 28 Aug 2025 16:36:58 -0700
Subject: [PATCH] remove convert in compiler_funcs (#13698)

Summary:

Directly use convert_pt2e since we've moved custom handling out

Reviewed By: zonglinpeng

Differential Revision: D81065417
---
 backends/cadence/aot/compiler.py       |  4 ++--
 backends/cadence/aot/compiler_funcs.py | 11 +----------
 2 files changed, 3 insertions(+), 12 deletions(-)

diff --git a/backends/cadence/aot/compiler.py b/backends/cadence/aot/compiler.py
index eaabc6589b5..5b7aef3c129 100644
--- a/backends/cadence/aot/compiler.py
+++ b/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:")
diff --git a/backends/cadence/aot/compiler_funcs.py b/backends/cadence/aot/compiler_funcs.py
index 5d5523ba31d..6ff6057255c 100644
--- a/backends/cadence/aot/compiler_funcs.py
+++ b/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