Skip to content

Arm backend: Delegate constant index select on U55 - #22605

Open
perheld wants to merge 1 commit into
pytorch:mainfrom
perheld:change-1320666
Open

Arm backend: Delegate constant index select on U55#22605
perheld wants to merge 1 commit into
pytorch:mainfrom
perheld:change-1320666

Conversation

@perheld

@perheld perheld commented Sep 8, 2026

Copy link
Copy Markdown
Collaborator

Lower constant index_select operations to U55-supported ops.

Contiguous indices such as [1, 2, 3] become one slice: [1:4]. Indices such as [1, 3, 1] become one-element slices followed by concatenation. This supports noncontiguous, reordered, and repeated indices.

Preserve CPU fallback for runtime, empty, and symbolic cases.

Authored with Codex.

Change-Id: I1073680c7844925f905bd3d2de2b47fc3a0d62c1

cc @digantdesai @freddan80 @per @zingo @oscarandersson8218 @mansnils @Sebastian-Larsson @robell @rascani

@perheld
perheld requested a review from digantdesai as a code owner September 8, 2026 15:22
@perheld perheld added partner: arm For backend delegation, kernels, demo, etc. from the 3rd-party partner, Arm ciflow/trunk release notes: arm Changes to the ARM backend delegate labels Sep 8, 2026
@pytorch-bot

pytorch-bot Bot commented Sep 8, 2026

Copy link
Copy Markdown

🔗 Helpful Links

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

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

❌ 1 New Failure

As of commit 0d9335d with merge base 7dc8641 (image):

NEW FAILURE - The following job has failed:

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 Sep 8, 2026
@github-actions github-actions Bot added the module: arm Issues related to arm backend label Sep 8, 2026
@perheld
perheld requested a review from zingo September 8, 2026 15:26
Lower constant index_select operations to U55-supported ops.

Contiguous indices such as [1, 2, 3] become one slice: [1:4].
Indices such as [1, 3, 1] become one-element slices followed by
concatenation. This supports noncontiguous, reordered, and repeated
indices.

Preserve CPU fallback for runtime, empty, and symbolic cases.

Authored with Codex.

Change-Id: I1073680c7844925f905bd3d2de2b47fc3a0d62c1
Signed-off-by: Per Held <per.held@arm.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ciflow/trunk CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. module: arm Issues related to arm backend partner: arm For backend delegation, kernels, demo, etc. from the 3rd-party partner, Arm release notes: arm Changes to the ARM backend delegate

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants