Skip to content

Arm backend: Add infer_shapes to test infra.#20422

Merged
oscarandersson8218 merged 2 commits into
pytorch:mainfrom
oscarandersson8218:infer_shapes_support
Jun 23, 2026
Merged

Arm backend: Add infer_shapes to test infra.#20422
oscarandersson8218 merged 2 commits into
pytorch:mainfrom
oscarandersson8218:infer_shapes_support

Conversation

@oscarandersson8218

@oscarandersson8218 oscarandersson8218 commented Jun 22, 2026

Copy link
Copy Markdown
Collaborator

For tosa-flatbuffers with unresolved shapes, we need to invoke infer_shapes to resolve shapes before invoking tosa_reference_model. This change will make sure that infer_shapes is called before reference_model when shape-extension is enabled and any input has a symbolic dimension.

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

For tosa-flatbuffers with unresolved shapes, we need to invoke
infer_shapes to resolve shapes before invoking tosa_reference_model.
This change will make sure that infer_shapes is called before
reference_model when shape-extension is enabled and any input has a
symbolic dimension.

Signed-off-by: Oscar Andersson <oscar.andersson@arm.com>
Change-Id: I3a9aa6fe8013e319fa974b531ad44afd9b1ce829
@pytorch-bot

pytorch-bot Bot commented Jun 22, 2026

Copy link
Copy Markdown

🔗 Helpful Links

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

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

❗ 1 Active SEVs

There are 1 currently active SEVs. If your PR is affected, please view them below:

❌ 2 New Failures, 1 Cancelled Job, 3 Unrelated Failures

As of commit 6565b98 with merge base 111e93c (image):

NEW FAILURES - The following jobs have failed:

CANCELLED JOB - The following job was cancelled. Please retry:

FLAKY - The following job failed but was likely due to flakiness present on trunk:

BROKEN TRUNK - The following jobs failed but was present on the merge base:

👉 Rebase onto the `viable/strict` branch to avoid these failures

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 Jun 22, 2026
@github-actions github-actions Bot added ciflow/trunk module: arm Issues related to arm backend labels Jun 22, 2026
@oscarandersson8218 oscarandersson8218 added partner: arm For backend delegation, kernels, demo, etc. from the 3rd-party partner, Arm release notes: none Do not include this in the release notes labels Jun 22, 2026
@oscarandersson8218

Copy link
Copy Markdown
Collaborator Author

Unrelated failures.

@oscarandersson8218 oscarandersson8218 merged commit 1c2000f into pytorch:main Jun 23, 2026
484 of 492 checks passed
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: none Do not include this in the release notes

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants