fix the expand_copy lower issue

Differential Revision: D69470884

Pull Request resolved: #8380

Author: billmguo

backends/qualcomm/_passes/convert_bmm_to_matmul.py

@@ -46,7 +46,8 @@ def _get_ordered_inputs(
     def call(self, graph_module: torch.fx.GraphModule):
         graph = graph_module.graph
         partitions = get_source_partitions(
-            graph, [operator.matmul, torch.matmul, torch.bmm]
+            graph,
+            [operator.matmul, torch.matmul, torch.bmm, torch.ops.aten.matmul.default],
         )
         for _, src_partitions in partitions.items():
             for src_partition in src_partitions:

backends/qualcomm/_passes/convert_to_linear.py

@@ -190,7 +190,9 @@ def _extract_bmm_ops(self, partitioned_nodes: List[edge_op]) -> List[torch.fx.No
         return ret
 
     def _convert(self, graph_module: torch.fx.GraphModule):
-        partitions = get_source_partitions(graph_module.graph, [torch.nn.Linear])
+        partitions = get_source_partitions(
+            graph_module.graph, [torch.nn.Linear, torch.ops.aten.linear.default]
+        )
         for _, src_partitions in partitions.items():
             for src_partition in src_partitions:
                 op_cnt = Counter(