Skip to content

Cortex-M: share layout-aware CMSIS-NN kernels - #21964

Draft
rascani wants to merge 9 commits into
gh/rascani/9/headfrom
gh/rascani/10/head
Draft

Cortex-M: share layout-aware CMSIS-NN kernels#21964
rascani wants to merge 9 commits into
gh/rascani/9/headfrom
gh/rascani/10/head

Conversation

@rascani

@rascani rascani commented Aug 20, 2026

Copy link
Copy Markdown
Contributor

Refactor convolution, depthwise convolution, transpose convolution, average
pooling, and max pooling behind shared internal implementations that accept a
private activation-layout mode.

Existing registered operators continue selecting the legacy NCHW-logical
contract over channels-last storage. This commit changes no public schema and no
AOT lowering; it provides the common implementation the following fixed-NHWC
wrappers use.

Local Corstone-300 run of the full Cortex-M suite, including the default-mode
dialect tests and MobileNetV3 lowering, and lintrunner.

Authored with Codex.

[ghstack-poisoned]
@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/21964

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

✅ You can merge normally! (6 Unrelated Failures)

As of commit 069bec4 with merge base 8b93850 (image):

BROKEN TRUNK - The following jobs 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.

[ghstack-poisoned]
[ghstack-poisoned]
[ghstack-poisoned]
[ghstack-poisoned]
[ghstack-poisoned]
[ghstack-poisoned]
[ghstack-poisoned]
Base automatically changed from gh/rascani/19/head to gh/rascani/9/head August 21, 2026 20:53
[ghstack-poisoned]
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.

1 participant