Skip to content

Arm backend: Add fused i8 warp+downsample example - #21972

Merged
oscarandersson8218 merged 1 commit into
pytorch:mainfrom
oscarandersson8218:fusion_custom_example
Aug 20, 2026
Merged

Arm backend: Add fused i8 warp+downsample example#21972
oscarandersson8218 merged 1 commit into
pytorch:mainfrom
oscarandersson8218:fusion_custom_example

Conversation

@oscarandersson8218

@oscarandersson8218 oscarandersson8218 commented Aug 20, 2026

Copy link
Copy Markdown
Collaborator

Adds example where int8 warp+downsample is fused into one custom shader. This pattern is present in rife model and such a fusion can reduce memory bandwidth significantly as it avoids processing pixels that are not used by downsampling.

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

Adds example where int8 warp+downsample is fused into one custom shader.
This pattern is present in rife model and such a fusion can reduce
memory bandwidth significantly as it avoids processing pixels that are
not used by downsampling.

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

pytorch-bot Bot commented Aug 20, 2026

Copy link
Copy Markdown

🔗 Helpful Links

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

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

❌ 1 Cancelled Job, 1 Unrelated Failure

As of commit 14f33be with merge base 92079a2 (image):

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

BROKEN TRUNK - The following job failed but were 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 Aug 20, 2026
@github-actions github-actions Bot added ciflow/trunk module: arm Issues related to arm backend labels Aug 20, 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 Aug 20, 2026
@zingo

zingo commented Aug 20, 2026

Copy link
Copy Markdown
Collaborator

nit: Maybe also update examples/arm/QAT_example/README.md

@zingo zingo left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I assume the bin files are used/needed and in this stage not generated?

@oscarandersson8218

Copy link
Copy Markdown
Collaborator Author

I assume the bin files are used/needed and in this stage not generated?

They are needed just like the ones in https://github.com/pytorch/executorch/tree/main/backends/arm/vgf/shaders.

nit: Maybe also update examples/arm/QAT_example/README.md

I'll add this in a follow up if that's ok.

@oscarandersson8218
oscarandersson8218 merged commit 8362c00 into pytorch:main Aug 20, 2026
529 of 543 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