[MPS Delegate] Add new mps runtime with support for FP16, lifted and unlifted graphs (iOS15+, macOS12+)#1655
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/1655
Note: Links to docs will display an error until the docs builds have been completed. ✅ You can merge normally! (1 Unrelated Failure)As of commit 8d94299 with merge base 93dd96f ( BROKEN TRUNK - The following job 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. |
|
@shoumikhin has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator. |
5 similar comments
|
@shoumikhin has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator. |
|
@shoumikhin has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator. |
|
@shoumikhin has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator. |
|
@shoumikhin has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator. |
|
@shoumikhin has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator. |
345c198 to
923d30a
Compare
|
@shoumikhin has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator. |
|
@shoumikhin has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator. |
1 similar comment
|
@shoumikhin has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator. |
|
@shoumikhin merged this pull request in cbc0c95. |
This PR changes the MPS Backend runtime to support for iOS15+/macOS12+ (previous runtime was limited to iOS17/macOS14 only). Additionally, this PR contains changes such as support for both lifted and unlifted graphs, support for torch.export API, optimizations for FP16 and faster model loading during runtime (more information in the summary).
Summary of changes:
iOS17/macOSSonoma APIs for serialization of the MPSGraphExecutable. Instead of creating the MPSGraph nodes and serializing them during AOT, create the corresponding entries of the EdgeIR nodes in the FlatBuffer, and parse them during runtime to construct the graph. This removes any dependency on iOS17 / macOS 14.0 APIs.MPSGraphBuilderclass and corresponding ops fromoperators/folder.Models summary: