Replace edge.aten.clone with edge.dim_order_ops._clone_dim_order

Author: keyprocedure

backends/qualcomm/_passes/convert_bmm_to_matmul.py

@@ -21,7 +21,7 @@ class ConvertBmmToMatmul(ExportPass):
 
     view_copy = exir_ops.edge.aten.view_copy.default
     expand_copy = exir_ops.edge.aten.expand_copy.default
-    clone = exir_ops.edge.aten.clone.default
+    clone = exir_ops.edge.dim_order_ops._clone_dim_order.default
     bmm = exir_ops.edge.aten.bmm.default
     matmul = exir_ops.edge.aten.matmul.default
     patterns = [

backends/qualcomm/_passes/remove_redundancy.py

@@ -19,7 +19,7 @@ def __init__(self, quantization_capture=False):
         self.redundant_ops_general = {
             torch.clone: self._default_condition,
             torch.ops.aten.clone.default: self._default_condition,
-            exir_ops.edge.aten.clone.default: self._default_condition,
+            exir_ops.edge.dim_order_ops._clone_dim_order.default: self._default_condition,
             torch.ops.aten.alias.default: self._default_condition,
             exir_ops.edge.aten.alias.default: self._default_condition,
             exir_ops.edge.aten.alias_copy.default: self._default_condition,

backends/qualcomm/partition/common_defs.py

@@ -10,7 +10,7 @@
 from executorch.exir.dialects._ops import ops as exir_ops
 
 not_supported_operator = [
-    exir_ops.edge.aten.clone.default,
+    exir_ops.edge.dim_order_ops._clone_dim_order.default,
     exir_ops.edge.quantized_decomposed.embedding_4bit.dtype,
 ]
 

backends/qualcomm/tests/utils.py

@@ -660,7 +660,7 @@ def _insert_clone(
                         users = list(node.users.keys())
                         inserted_node = graph_module.graph.create_node(
                             "call_function",
-                            exir_ops.edge.aten.clone.default,
+                            exir_ops.edge.dim_order_ops._clone_dim_order.default,
                             (node,),
                         )
                         inserted_node.meta["val"] = node.meta["val"]