Skip to content

Add multi-method support for ETRecord edge dialect programs#17185

Open
larryliu0820 wants to merge 1 commit intomainfrom
export-D92215098
Open

Add multi-method support for ETRecord edge dialect programs#17185
larryliu0820 wants to merge 1 commit intomainfrom
export-D92215098

Conversation

@larryliu0820
Copy link
Contributor

Summary:
ETRecord previously only supported a single edge dialect program (defaulting to
the "forward" method). This change adds support for multiple methods like
vision_encoder, token_embedding, and text_decoder that are used in VLM exports.

The edge_dialect_program field can now be either a single ExportedProgram (for
backward compatibility) or a Dict[str, ExportedProgram] mapping method names to
their exported programs.

Key changes:

  • Updated ETRecord to accept Dict[str, ExportedProgram] for edge_dialect_program
  • Modified _process_edge_dialect_program to extract all methods from EdgeProgramManager
  • Updated save/parse logic to handle multi-method format with backward compatibility
  • New format uses paths like edge_dialect_exported_program/{method_name}

Authored with Claude Code.

Differential Revision: D92215098

@pytorch-bot
Copy link

pytorch-bot bot commented Feb 4, 2026

🔗 Helpful Links

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

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

❌ 7 New Failures

As of commit b1c12b6 with merge base c0cef6c (image):

NEW FAILURES - The following jobs have failed:

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 Feb 4, 2026
@meta-codesync
Copy link
Contributor

meta-codesync bot commented Feb 4, 2026

@larryliu0820 has exported this pull request. If you are a Meta employee, you can view the originating Diff in D92215098.

@larryliu0820 larryliu0820 added the release notes: devtools Changes to dev tooling, for example the debugger & profiler label Feb 4, 2026
Summary:
ETRecord previously only supported a single edge dialect program (defaulting to
the "forward" method). This change adds support for multiple methods like
vision_encoder, token_embedding, and text_decoder that are used in VLM exports.

The edge_dialect_program field can now be either a single ExportedProgram (for
backward compatibility) or a Dict[str, ExportedProgram] mapping method names to
their exported programs.

Key changes:
- Updated ETRecord to accept Dict[str, ExportedProgram] for edge_dialect_program
- Modified _process_edge_dialect_program to extract all methods from EdgeProgramManager
- Updated save/parse logic to handle multi-method format with backward compatibility
- New format uses paths like edge_dialect_exported_program/{method_name}

Authored with Claude Code.

Differential Revision: D92215098
@meta-codesync
Copy link
Contributor

meta-codesync bot commented Feb 4, 2026

@larryliu0820 has imported this pull request. If you are a Meta employee, you can view this in D92215098.

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. fb-exported meta-exported release notes: devtools Changes to dev tooling, for example the debugger & profiler

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants