Skip to content

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

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

rascani wants to merge 14 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.

❌ 4 Cancelled Jobs, 1 Unrelated Failure

As of commit ad3fe3a with merge base 469debd (image):

CANCELLED JOBS - The following jobs were cancelled. Please retry:

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

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

This branch was previously deployed

1 inactive deployment
cadence — ad3fe3a0 Deployed Aug 25, 2026 by rascani
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