fix: Improve partitioning + lowering systems in torch.compile path#1879
Merged
gs-olive merged 2 commits intopytorch:mainfrom May 19, 2023
Merged
fix: Improve partitioning + lowering systems in torch.compile path#1879gs-olive merged 2 commits intopytorch:mainfrom
torch.compile path#1879gs-olive merged 2 commits intopytorch:mainfrom
Conversation
gs-olive
added
the
component: dynamo
gs-olive
force-pushed
the
dynamo_partitoner_lowering
branch
3 times, most recently
from
f745135 to
fc63b2c
Compare
gs-olive
force-pushed
the
dynamo_partitoner_lowering
branch
from
fc63b2c to
0d85466
Compare
gs-olive
force-pushed
the
dynamo_partitoner_lowering
branch
3 times, most recently
from
c3f88c1 to
a77017c
Compare
This was referenced May 15, 2023
- Improve torch.compile Dynamo partitioning system by incorporating key arguments including `min_block_size` and `torch_executed_ops`, which are available for use in TorchScript - Improve torch.compile lowering system by adding key new decompositions to improve coverage and reduce the number of unique operators requiring implementation - Update testing framework to use utilities, reducing code replication - Add extensive testing of new partitioning system and lowering phases
gs-olive
force-pushed
the
dynamo_partitoner_lowering
branch
2 times, most recently
from
4f527a0 to
5d77b6b
Compare
- Update default `min_block_size`
gs-olive
force-pushed
the
dynamo_partitoner_lowering
branch
from
5d77b6b to
357d18d
Compare
gs-olive
added a commit
that referenced
this pull request
May 19, 2023
narendasan
pushed a commit
that referenced
this pull request
May 30, 2023
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
3 participants
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Description
min_block_sizeandtorch_executed_ops, which are available for use in TorchScriptNote: Not to be merged prior to
release/1.4branch finalization (features are experimental and evolving)Fixes #1832
Type of change
Checklist: