fix: Improve partitioning + lowering systems in `torch.compile` path by gs-olive · Pull Request #1879 · pytorch/TensorRT

@gs-olive self-assigned this

May 2, 2023

narendasan

@gs-olive gs-olive changed the base branch from dynamo_changes to main

May 3, 2023 00:14

This was referenced

May 15, 2023

@gs-olive

- 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

- Update default `min_block_size`

@gs-olive gs-olive deleted the dynamo_partitoner_lowering branch

May 19, 2023 01:24

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