Add ability to specify CoreML pipeline passes #16118
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/16118
Note: Links to docs will display an error until the docs builds have been completed. ✅ No FailuresAs of commit d9174e4 with merge base 9193566 ( This comment was automatically generated by Dr. CI and updates every 15 minutes. |
meta-cla
bot
added
the
CLA Signed
There was a problem hiding this comment.
Pull request overview
This PR adds the ability to specify a custom CoreML pass pipeline through the CompileSpec system, enabling users to customize the model compilation process beyond the default pipeline.
Key Changes:
- Added
PASS_PIPELINEkey to the compile spec enumeration - Implemented methods to serialize/deserialize pass pipeline configurations
- Integrated custom pass pipeline into the CoreML conversion process
💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.
|
@pytorchbot label "release notes: none" |
pytorch-bot
bot
added
the
release notes: none
jgibson2
changed the title
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
There was a problem hiding this comment.
Pull request overview
Copilot reviewed 1 out of 1 changed files in this pull request and generated 3 comments.
💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.
1 participant
Summary
Adds the ability to specify a set of CoreML passes as a CompileSpec, allowing additional customization of the model compilation.
Test plan
Converted a model and made sure it worked with a custom pipeline. Also ensured via print statements that the passes were translated correctly.