Skip to content

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

Merged
SS-JIA merged 1 commit into
gh/SS-JIA/500/origfrom
gh/SS-JIA/501/orig
Mar 24, 2026
Merged

[ET-VK] Support alias_copy in partitioner by removing it as a redundant op#18461
SS-JIA merged 1 commit into
gh/SS-JIA/500/origfrom
gh/SS-JIA/501/orig

Conversation

@pytorchbot

Copy link
Copy Markdown
Collaborator

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
@diff-train-skip-merge

…nt op

Models using QK normalization (e.g., Qwen3) generate aten.alias_copy.default
nodes from RMSNorm. Without Vulkan support for this op, these nodes create
partition boundaries that cause SpecViolationError during export. Since
alias_copy is a no-op identity operation, the fix registers it in op_registry
so the partitioner delegates it, and adds it to RemoveRedundantOpsTransform
so preprocess removes it before serialization. The C++ runtime never needs to
handle alias_copy.

Differential Revision: [D97757921](https://our.internmc.facebook.com/intern/diff/D97757921/)

ghstack-source-id: 356136429
Pull Request resolved: #18408
@pytorchbot pytorchbot requested a review from SS-JIA as a code owner March 24, 2026 19:59
@pytorch-bot

pytorch-bot Bot commented Mar 24, 2026

Copy link
Copy Markdown

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/18461

Note: Links to docs will display an error until the docs builds have been completed.

This comment was automatically generated by Dr. CI and updates every 15 minutes.

@meta-cla meta-cla Bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Mar 24, 2026
@SS-JIA SS-JIA merged commit 186ad12 into gh/SS-JIA/500/orig Mar 24, 2026
133 of 141 checks passed
@SS-JIA SS-JIA deleted the gh/SS-JIA/501/orig branch March 24, 2026 20:05
SS-JIA pushed a commit that referenced this pull request Mar 24, 2026
…nt 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>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants