[ET-VK] Support alias_copy in partitioner by removing it as a redundant op (#18461)

This PR was created by the merge bot to help merge the original PR into
the main branch.
ghstack PR number: #18408 by
@SS-JIA
^ Please use this as the source of truth for the PR details, comments,
and reviews
ghstack PR base:
https://github.com/pytorch/executorch/tree/gh/SS-JIA/501/base
ghstack PR head:
https://github.com/pytorch/executorch/tree/gh/SS-JIA/501/head
Merge bot PR base:
https://github.com/pytorch/executorch/tree/gh/SS-JIA/500/orig
Merge bot PR head:
https://github.com/pytorch/executorch/tree/gh/SS-JIA/501/orig
Differential Revision:
[D97757921](https://our.internmc.facebook.com/intern/diff/D97757921/)
@diff-train-skip-merge

Co-authored-by: ssjia <ssjia@devvm26340.ftw0.facebook.com>

Author: 2 people

backends/vulkan/_passes/remove_redundant_ops.py

@@ -28,6 +28,8 @@ class RemoveRedundantOpsTransform(ExportPass):
         exir_ops.edge.aten.clone.default,
         torch.ops.aten.alias.default,
         exir_ops.edge.aten.alias.default,
+        torch.ops.aten.alias_copy.default,
+        exir_ops.edge.aten.alias_copy.default,
         exir_ops.edge.aten.lift_fresh_copy.default,
         exir_ops.edge.dim_order_ops._to_dim_order_copy.default,
         exir_ops.edge.dim_order_ops._clone_dim_order.default,

backends/vulkan/op_registry.py

@@ -1125,6 +1125,20 @@ def register_clone_dim_order():
     )
 
 
+# alias_copy is a no-op identity operation (same as clone/alias). It is removed
+# by RemoveRedundantOpsTransform during preprocess, so the C++ runtime never sees
+# it. Registering it here ensures the partitioner delegates it to Vulkan rather
+# than creating partition boundaries that break the graph.
+@update_features(exir_ops.edge.aten.alias_copy.default)
+def register_alias_copy():
+    return OpFeatures(
+        inputs_storage=utils.ANY_STORAGE,
+        inputs_dtypes=utils.FP_INT_BOOL_T,
+        supports_resize=True,
+        supports_highdim=True,
+    )
+
+
 # =============================================================================
 # Gather.cpp
 # =============================================================================