Qualcomm AI Engine Direct - Fix for documentation and reuse dead_code_elimination_pass (#18644)

zhaoxul-qti · web-flow · commit 7f02bd740cc9 · 2026-04-07T15:10:27.000-05:00
diff --git a/backends/qualcomm/_passes/canonicalize_conv.py b/backends/qualcomm/_passes/canonicalize_conv.py
@@ -10,6 +10,7 @@
 
 from executorch.backends.qualcomm.builders.utils import get_parameter, set_parameter
 from executorch.exir.pass_base import ExportPass, PassResult
+from executorch.exir.passes import dead_code_elimination_pass
 from torch._guards import detect_fake_mode
 
 from .utils import append_qdq, copy_meta
@@ -199,6 +200,5 @@ def call(self, graph_module: torch.fx.GraphModule):
                 for user in node.users.copy():
                     user.replace_input_with(node, squeeze_node)
 
-        graph.eliminate_dead_code()
-        graph_module.recompile()
+        dead_code_elimination_pass(graph_module)
         return PassResult(graph_module, True)
diff --git a/backends/qualcomm/_passes/convert_bmm_to_matmul.py b/backends/qualcomm/_passes/convert_bmm_to_matmul.py
@@ -10,6 +10,7 @@
 import torch
 from executorch.exir.dialects._ops import ops as exir_ops
 from executorch.exir.pass_base import ExportPass, PassResult
+from executorch.exir.passes import dead_code_elimination_pass
 from torch.fx.passes.utils.source_matcher_utils import get_source_partitions
 
 
@@ -78,6 +79,5 @@ def call(self, graph_module: torch.fx.GraphModule):
                     for user in output.users.copy():
                         user.replace_input_with(output, matmul_node)
 
-        graph.eliminate_dead_code()
-        graph_module.recompile()
+        dead_code_elimination_pass(graph_module)
         return PassResult(graph_module, True)
diff --git a/backends/qualcomm/_passes/convert_linear_to_conv2d.py b/backends/qualcomm/_passes/convert_linear_to_conv2d.py
@@ -8,6 +8,7 @@
 from executorch.backends.qualcomm._passes.utils import append_qdq, copy_meta
 from executorch.backends.qualcomm.builders.utils import get_parameter, set_parameter
 from executorch.exir.pass_base import ExportPass, PassResult
+from executorch.exir.passes import dead_code_elimination_pass
 from torch.fx import GraphModule
 from torchao.quantization.pt2e.utils import get_new_attr_name_with_prefix
 
@@ -227,6 +228,5 @@ def call(self, graph_module: GraphModule):
                     node.replace_all_uses_with(y)
                     graph.erase_node(node)
 
-        graph.eliminate_dead_code()
-        graph_module.recompile()
+        dead_code_elimination_pass(graph_module)
         return PassResult(graph_module, True)
diff --git a/backends/qualcomm/_passes/convert_square_to_pow.py b/backends/qualcomm/_passes/convert_square_to_pow.py
@@ -5,6 +5,7 @@
 # LICENSE file in the root directory of this source tree.
 import torch
 from executorch.exir.pass_base import ExportPass, PassResult
+from executorch.exir.passes import dead_code_elimination_pass
 
 from .utils import copy_meta
 
@@ -33,6 +34,5 @@ def call(self, graph_module: torch.fx.GraphModule) -> PassResult:
                 for user in node.users.copy():
                     user.replace_input_with(node, pow_node)
 
-        graph.eliminate_dead_code()
-        graph_module.recompile()
+        dead_code_elimination_pass(graph_module)
         return PassResult(graph_module, True)
diff --git a/backends/qualcomm/_passes/decompose_any.py b/backends/qualcomm/_passes/decompose_any.py
@@ -7,6 +7,7 @@
 import torch
 from executorch.exir import to_edge
 from executorch.exir.pass_base import ExportPass, PassResult
+from executorch.exir.passes import dead_code_elimination_pass
 
 from .utils import merge_decomposed_graph
 
@@ -59,6 +60,5 @@ def call(self, graph_module: torch.fx.GraphModule) -> PassResult:
                     )
                     graph.erase_node(node)
 
-        graph.eliminate_dead_code()
-        graph_module.recompile()
+        dead_code_elimination_pass(graph_module)
         return PassResult(graph_module, True)
diff --git a/backends/qualcomm/_passes/decompose_binary_alpha.py b/backends/qualcomm/_passes/decompose_binary_alpha.py
@@ -6,6 +6,7 @@
 
 import torch
 from executorch.exir.pass_base import ExportPass, PassResult
+from executorch.exir.passes import dead_code_elimination_pass
 
 from .utils import copy_meta
 
@@ -56,6 +57,5 @@ def call(self, graph_module: torch.fx.GraphModule) -> PassResult:
                         mul_node,
                     )
 
-        graph.eliminate_dead_code()
-        graph_module.recompile()
+        dead_code_elimination_pass(graph_module)
         return PassResult(graph_module, True)
diff --git a/backends/qualcomm/_passes/decompose_cdist.py b/backends/qualcomm/_passes/decompose_cdist.py
@@ -6,6 +6,7 @@
 
 import torch
 from executorch.exir.pass_base import ExportPass, PassResult
+from executorch.exir.passes import dead_code_elimination_pass
 
 from .utils import merge_decomposed_graph
 
@@ -64,6 +65,5 @@ def call(self, graph_module: torch.fx.GraphModule) -> PassResult:
                     )
                     graph.erase_node(node)
 
-        graph.eliminate_dead_code()
-        graph_module.recompile()
+        dead_code_elimination_pass(graph_module)
         return PassResult(graph_module, True)
diff --git a/backends/qualcomm/_passes/decompose_col_im.py b/backends/qualcomm/_passes/decompose_col_im.py
@@ -6,6 +6,7 @@
 import torch
 from executorch.exir.dialects._ops import ops as exir_ops
 from executorch.exir.pass_base import ExportPass, PassResult
+from executorch.exir.passes import dead_code_elimination_pass
 
 from .utils import copy_meta
 
@@ -117,5 +118,5 @@ def _decompose_col2im(self, graph_module: torch.fx.GraphModule):
     def call(self, graph_module: torch.fx.GraphModule):
         self._decompose_im2col(graph_module)
         self._decompose_col2im(graph_module)
-        graph_module.recompile()
+        dead_code_elimination_pass(graph_module)
         return PassResult(graph_module, True)
diff --git a/backends/qualcomm/_passes/decompose_einsum.py b/backends/qualcomm/_passes/decompose_einsum.py
@@ -6,6 +6,7 @@
 
 import torch
 from executorch.exir.pass_base import ExportPass, PassResult
+from executorch.exir.passes import dead_code_elimination_pass
 from torch.fx.experimental.proxy_tensor import make_fx
 
 from .utils import merge_decomposed_graph
@@ -46,6 +47,5 @@ def call(self, graph_module: torch.fx.GraphModule) -> PassResult:
                     )
                     graph.erase_node(node)
 
-        graph.eliminate_dead_code()
-        graph_module.recompile()
+        dead_code_elimination_pass(graph_module)
         return PassResult(graph_module, True)
diff --git a/backends/qualcomm/_passes/decompose_expm1.py b/backends/qualcomm/_passes/decompose_expm1.py
@@ -6,6 +6,7 @@
 
 import torch
 from executorch.exir.pass_base import ExportPass, PassResult
+from executorch.exir.passes import dead_code_elimination_pass
 
 from .utils import copy_meta
 
@@ -41,6 +42,5 @@ def call(self, graph_module: torch.fx.GraphModule) -> PassResult:
                 for user in node.users.copy():
                     user.replace_input_with(node, sub_node)
 
-        graph.eliminate_dead_code()
-        graph_module.recompile()
+        dead_code_elimination_pass(graph_module)
         return PassResult(graph_module, True)
diff --git a/backends/qualcomm/_passes/decompose_floor_divide.py b/backends/qualcomm/_passes/decompose_floor_divide.py
@@ -6,6 +6,7 @@
 
 import torch
 from executorch.exir.pass_base import ExportPass, PassResult
+from executorch.exir.passes import dead_code_elimination_pass
 
 from .utils import merge_decomposed_graph
 
@@ -57,6 +58,5 @@ def call(self, graph_module: torch.fx.GraphModule) -> PassResult:
                     )
                     graph.erase_node(node)
 
-        graph.eliminate_dead_code()
-        graph_module.recompile()
+        dead_code_elimination_pass(graph_module)
         return PassResult(graph_module, True)
diff --git a/backends/qualcomm/_passes/decompose_glu.py b/backends/qualcomm/_passes/decompose_glu.py
@@ -6,6 +6,7 @@
 
 import torch
 from executorch.exir.pass_base import ExportPass, PassResult
+from executorch.exir.passes import dead_code_elimination_pass
 
 from .utils import merge_decomposed_graph
 
@@ -50,6 +51,5 @@ def call(self, graph_module: torch.fx.GraphModule) -> PassResult:
                     )
                     graph.erase_node(node)
 
-        graph.eliminate_dead_code()
-        graph_module.recompile()
+        dead_code_elimination_pass(graph_module)
         return PassResult(graph_module, True)
diff --git a/backends/qualcomm/_passes/decompose_linalg_vector_norm.py b/backends/qualcomm/_passes/decompose_linalg_vector_norm.py
@@ -7,6 +7,7 @@
 import torch
 from executorch.exir import to_edge
 from executorch.exir.pass_base import ExportPass, PassResult
+from executorch.exir.passes import dead_code_elimination_pass
 
 from .utils import merge_decomposed_graph
 
@@ -70,6 +71,5 @@ def call(self, graph_module: torch.fx.GraphModule) -> PassResult:
                     )
                     graph.erase_node(node)
 
-        graph.eliminate_dead_code()
-        graph_module.recompile()
+        dead_code_elimination_pass(graph_module)
         return PassResult(graph_module, True)
diff --git a/backends/qualcomm/_passes/decompose_maxpool3d.py b/backends/qualcomm/_passes/decompose_maxpool3d.py
@@ -10,6 +10,7 @@
 import torch.nn as nn
 from executorch.exir import to_edge
 from executorch.exir.pass_base import ExportPass, PassResult
+from executorch.exir.passes import dead_code_elimination_pass
 
 from .utils import merge_decomposed_graph
 
@@ -128,6 +129,5 @@ def call(self, graph_module: torch.fx.GraphModule) -> PassResult:
                     )
                     graph.erase_node(node)
 
-        graph.eliminate_dead_code()
-        graph_module.recompile()
+        dead_code_elimination_pass(graph_module)
         return PassResult(graph_module, True)
diff --git a/backends/qualcomm/_passes/decompose_minmaxdim.py b/backends/qualcomm/_passes/decompose_minmaxdim.py
@@ -11,6 +11,7 @@
 import torch
 from executorch.exir.dialects._ops import ops as exir_ops
 from executorch.exir.pass_base import ExportPass, PassResult
+from executorch.exir.passes import dead_code_elimination_pass
 from torch.fx.passes.utils.source_matcher_utils import get_source_partitions
 
 
@@ -104,6 +105,5 @@ def call(self, graph_module: torch.fx.GraphModule):
                     for user in list(idx_node.users):
                         user.replace_input_with(idx_node, argmin_node)
 
-        graph.eliminate_dead_code()
-        graph_module.recompile()
+        dead_code_elimination_pass(graph_module)
         return PassResult(graph_module, True)
diff --git a/backends/qualcomm/_passes/decompose_roll.py b/backends/qualcomm/_passes/decompose_roll.py
@@ -6,6 +6,7 @@
 import torch
 
 from executorch.exir.pass_base import ExportPass, PassResult
+from executorch.exir.passes import dead_code_elimination_pass
 
 from .utils import merge_decomposed_graph
 
@@ -73,6 +74,5 @@ def call(self, graph_module: torch.fx.GraphModule) -> PassResult:
                     )
                     graph.erase_node(node)
 
-        graph.eliminate_dead_code()
-        graph_module.recompile()
+        dead_code_elimination_pass(graph_module)
         return PassResult(graph_module, True)
diff --git a/backends/qualcomm/_passes/decompose_silu.py b/backends/qualcomm/_passes/decompose_silu.py
@@ -6,6 +6,7 @@
 
 import torch
 from executorch.exir.pass_base import ExportPass, PassResult
+from executorch.exir.passes import dead_code_elimination_pass
 
 from .utils import copy_meta
 
@@ -40,6 +41,5 @@ def call(self, graph_module: torch.fx.GraphModule):
                         for user in silu_node.users.copy():
                             user.replace_input_with(silu_node, mul_node)
 
-        graph.eliminate_dead_code()
-        graph_module.recompile()
+        dead_code_elimination_pass(graph_module)
         return PassResult(graph_module, True)
diff --git a/backends/qualcomm/_passes/decompose_threshold.py b/backends/qualcomm/_passes/decompose_threshold.py
@@ -6,6 +6,7 @@
 import torch
 
 from executorch.exir.pass_base import ExportPass, PassResult
+from executorch.exir.passes import dead_code_elimination_pass
 
 from .utils import merge_decomposed_graph
 
@@ -56,6 +57,5 @@ def call(self, graph_module: torch.fx.GraphModule) -> PassResult:
                     )
                     graph.erase_node(node)
 
-        graph.eliminate_dead_code()
-        graph_module.recompile()
+        dead_code_elimination_pass(graph_module)
         return PassResult(graph_module, True)
diff --git a/backends/qualcomm/_passes/decompose_triu.py b/backends/qualcomm/_passes/decompose_triu.py
@@ -8,6 +8,7 @@
 
 import torch
 from executorch.exir.pass_base import ExportPass, PassResult
+from executorch.exir.passes import dead_code_elimination_pass
 from torch._decomp import get_decompositions
 from torch.fx.experimental.proxy_tensor import make_fx
 
@@ -66,6 +67,5 @@ def call(self, graph_module: torch.fx.GraphModule) -> PassResult:
                     )
                     graph.erase_node(node)
 
-        graph.eliminate_dead_code()
-        graph_module.recompile()
+        dead_code_elimination_pass(graph_module)
         return PassResult(graph_module, True)
diff --git a/backends/qualcomm/_passes/decompose_wrap_with_autocast.py b/backends/qualcomm/_passes/decompose_wrap_with_autocast.py
@@ -9,6 +9,7 @@
 
 import torch
 from executorch.exir.pass_base import ExportPass, PassResult
+from executorch.exir.passes import dead_code_elimination_pass
 
 from .utils import merge_decomposed_graph
 
@@ -74,6 +75,5 @@ def _replace(self, gm: torch.fx.GraphModule) -> None:
 
     def call(self, graph_module: torch.fx.GraphModule) -> PassResult:
         self._replace(graph_module)
-        graph_module.graph.eliminate_dead_code()
-        graph_module.recompile()
+        dead_code_elimination_pass(graph_module)
         return PassResult(graph_module, True)
diff --git a/backends/qualcomm/_passes/expand_broadcast_tensor_shape.py b/backends/qualcomm/_passes/expand_broadcast_tensor_shape.py
@@ -62,6 +62,5 @@ def traverse_broadcast_node(self, graph_module: torch.fx.GraphModule):
 
     def call(self, graph_module: torch.fx.GraphModule):
         self.traverse_broadcast_node(graph_module)
-        graph_module.recompile()
         dead_code_elimination_pass(graph_module)
         return PassResult(graph_module, True)
diff --git a/backends/qualcomm/_passes/fixed_linear_keep_dim.py b/backends/qualcomm/_passes/fixed_linear_keep_dim.py
@@ -91,5 +91,4 @@ def _fixed_keep_dim(self, graph_module: torch.fx.GraphModule):
     def call(self, graph_module: torch.fx.GraphModule):
         self._fixed_keep_dim(graph_module)
         dead_code_elimination_pass(graph_module)
-        graph_module.recompile()
         return PassResult(graph_module, True)
diff --git a/backends/qualcomm/_passes/fold_qdq.py b/backends/qualcomm/_passes/fold_qdq.py
@@ -75,6 +75,5 @@ def _fold_q(self, graph_module: torch.fx.GraphModule) -> torch.fx.GraphModule:
     def call(self, graph_module: torch.fx.GraphModule):
         self._fold_dq(graph_module)
         self._fold_q(graph_module)
-        graph_module.recompile()
         dead_code_elimination_pass(graph_module)
         return PassResult(graph_module, True)
diff --git a/backends/qualcomm/_passes/fuse_consecutive_cast.py b/backends/qualcomm/_passes/fuse_consecutive_cast.py
@@ -111,6 +111,5 @@ def _fuse(self, graph_module: torch.fx.GraphModule) -> torch.fx.GraphModule:
     def call(self, graph_module: torch.fx.GraphModule):
         self._canonicalize_cast(graph_module)
         self._fuse(graph_module)
-        graph_module.recompile()
         dead_code_elimination_pass(graph_module)
         return PassResult(graph_module, True)
diff --git a/backends/qualcomm/_passes/fuse_consecutive_transpose.py b/backends/qualcomm/_passes/fuse_consecutive_transpose.py
@@ -104,6 +104,5 @@ def _fuse(self, graph_module: torch.fx.GraphModule) -> torch.fx.GraphModule:
     def call(self, graph_module: torch.fx.GraphModule):
         self._clone_transpose(graph_module)
         self._fuse(graph_module)
-        graph_module.recompile()
         dead_code_elimination_pass(graph_module)
         return PassResult(graph_module, True)
diff --git a/backends/qualcomm/_passes/insert_io_qdq.py b/backends/qualcomm/_passes/insert_io_qdq.py
@@ -20,6 +20,7 @@
 )
 from executorch.exir.dialects._ops import ops as exir_ops
 from executorch.exir.pass_base import ExportPass, PassResult
+from executorch.exir.passes import dead_code_elimination_pass
 
 
 class InsertIOQDQ(ExportPass):
@@ -156,6 +157,5 @@ def _insert(self, graph_module: torch.fx.GraphModule) -> torch.fx.GraphModule:
 
     def call(self, graph_module: torch.fx.GraphModule):
         self._insert(graph_module)
-        graph_module.graph.eliminate_dead_code()
-        graph_module.recompile()
+        dead_code_elimination_pass(graph_module)
         return PassResult(graph_module, True)
diff --git a/backends/qualcomm/_passes/insert_requantize.py b/backends/qualcomm/_passes/insert_requantize.py
@@ -17,6 +17,7 @@
 
 from executorch.exir.dialects._ops import ops as exir_ops
 from executorch.exir.pass_base import ExportPass, PassResult
+from executorch.exir.passes import dead_code_elimination_pass
 
 
 class InsertRequantize(ExportPass):
@@ -106,6 +107,5 @@ def _insert(self, graph_module: torch.fx.GraphModule) -> torch.fx.GraphModule:
 
     def call(self, graph_module: torch.fx.GraphModule):
         self._insert(graph_module)
-        graph_module.graph.eliminate_dead_code()
-        graph_module.recompile()
+        dead_code_elimination_pass(graph_module)
         return PassResult(graph_module, True)
diff --git a/backends/qualcomm/_passes/insert_reshape_for_reduce_ops.py b/backends/qualcomm/_passes/insert_reshape_for_reduce_ops.py
@@ -53,7 +53,6 @@ def call(self, graph_module: torch.fx.GraphModule):
                 modified = True
 
         if modified:
-            graph_module.recompile()
             dead_code_elimination_pass(graph_module)
 
         return PassResult(graph_module, modified)
diff --git a/backends/qualcomm/_passes/lift_constant_scalar_operands.py b/backends/qualcomm/_passes/lift_constant_scalar_operands.py
@@ -181,6 +181,5 @@ def _lift(self, gm: torch.fx.GraphModule) -> None:
 
     def call(self, graph_module: torch.fx.GraphModule):
         self._lift(graph_module)
-        graph_module.recompile()
         dead_code_elimination_pass(graph_module)
         return PassResult(graph_module, True)
diff --git a/backends/qualcomm/_passes/recompose_pad_maxpool2d.py b/backends/qualcomm/_passes/recompose_pad_maxpool2d.py
diff --git a/backends/qualcomm/_passes/recompose_pixel_unshuffle.py b/backends/qualcomm/_passes/recompose_pixel_unshuffle.py
diff --git a/backends/qualcomm/_passes/recompose_rms_norm.py b/backends/qualcomm/_passes/recompose_rms_norm.py
diff --git a/backends/qualcomm/_passes/reduce_dynamic_range.py b/backends/qualcomm/_passes/reduce_dynamic_range.py
diff --git a/backends/qualcomm/_passes/remove_0d_tensor.py b/backends/qualcomm/_passes/remove_0d_tensor.py
diff --git a/backends/qualcomm/_passes/remove_redundancy.py b/backends/qualcomm/_passes/remove_redundancy.py
diff --git a/backends/qualcomm/_passes/replace_arange_args.py b/backends/qualcomm/_passes/replace_arange_args.py
diff --git a/backends/qualcomm/debugger/format_outputs.py b/backends/qualcomm/debugger/format_outputs.py
diff --git a/examples/qualcomm/util_scripts/README.md b/examples/qualcomm/util_scripts/README.md
diff --git a/examples/qualcomm/util_scripts/cli.py b/examples/qualcomm/util_scripts/cli.py
diff --git a/examples/qualcomm/utils.py b/examples/qualcomm/utils.py
