to_edge_transform_and_lower

tarun292 commented 2 weeks ago

Summary: This diff introduces the to_edge_transform_and_lower API. The changes introduces are:

Adding support to the Parititioner class to register ops that it doesn't want to be composed
Changes to _program.py to add the implementation of to_edge_transform_and_lower()
Added a basic test case to test that Linear, SDPA & Linear + SDPA are not decomposed when asked and the corresponding backend consumes them.

Differential Revision: D56401086

pytorch-bot[bot] commented 2 weeks ago

:link: Helpful Links

:test_tube: See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/3483

:page_facing_up: Preview Python docs built from this PR

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

:x: 1 New Failure

As of commit 4ec669844253a2891f2803fd7d7dd0763ba3bbdd with merge base cc12d9b5aa346b0caeb29ab492b14639754ba4cf ():

NEW FAILURE - The following job has failed:

* [Lint / lintrunner / linux-job](https://hud.pytorch.org/pr/pytorch/executorch/3483#24520001210) ([gh](https://github.com/pytorch/executorch/actions/runs/8927193069/job/24520001210)) `>>> Lint for exir/program/test/test_program.py:`

This comment was automatically generated by Dr. CI and updates every 15 minutes.

facebook-github-bot commented 2 weeks ago

This pull request was exported from Phabricator. Differential Revision: D56401086

facebook-github-bot commented 2 weeks ago

This pull request was exported from Phabricator. Differential Revision: D56401086

facebook-github-bot commented 2 weeks ago

This pull request was exported from Phabricator. Differential Revision: D56401086

pytorch / executorch