Skip to content

examples: Add example usage scripts for torch_tensorrt.dynamo.compile path#1890

Closed
gs-olive wants to merge 2 commits intopytorch:mainfrom
gs-olive:dynamo_torch_compile_examples
Closed

examples: Add example usage scripts for torch_tensorrt.dynamo.compile path#1890
gs-olive wants to merge 2 commits intopytorch:mainfrom
gs-olive:dynamo_torch_compile_examples

Conversation

@gs-olive
Copy link
Contributor

@gs-olive gs-olive commented May 5, 2023

Description

  • Add sample scripts covering resnet18, transformers, and custom examples showcasing the torch_tensorrt.dynamo.compile path, which can compile models with data-dependent control flow and other such restrictions which can make other compilation methods more difficult
  • Cover different customizeable features allowed in the new backend

Addresses #1887

Type of change

  • Code samples

Checklist:

  • [ x ] My code follows the style guidelines of this project (You can use the linters)
  • [ x ] I have performed a self-review of my own code
  • [ x ] I have commented my code, particularly in hard-to-understand areas and hacks
  • [ x ] I have made corresponding changes to the documentation
  • [ x ] I have added tests to verify my fix or my feature
  • [ x ] New and existing unit tests pass locally with my changes
  • [ x ] I have added the relevant labels to my PR in so that relevant reviewers are notified

@gs-olive gs-olive added the component: dynamo Issues relating to the `torch.compile` or `torch._dynamo.export` paths label May 5, 2023
@gs-olive gs-olive self-assigned this May 5, 2023
@gs-olive gs-olive requested a review from narendasan May 5, 2023 17:13
@gs-olive gs-olive changed the title examples: Add example usage scripts for dynamo path examples: Add example usage scripts for torch_tensorrt.dynamo.torch_compile path May 5, 2023
@gs-olive gs-olive force-pushed the dynamo_torch_compile_examples branch from 9cbd31b to e6a1193 Compare May 16, 2023 03:22
@github-actions github-actions bot added the component: api [Python] Issues re: Python API label May 16, 2023
@gs-olive gs-olive added the WIP Work is in progress, pull request should not be merged yet label May 18, 2023
@gs-olive gs-olive changed the title examples: Add example usage scripts for torch_tensorrt.dynamo.torch_compile path examples: Add example usage scripts for torch_tensorrt.dynamo.dynamo_compile path May 18, 2023
@gs-olive gs-olive force-pushed the dynamo_torch_compile_examples branch from e6a1193 to 35b3f7d Compare May 18, 2023 22:45
@github-actions github-actions bot added the documentation Improvements or additions to documentation label May 18, 2023
@gs-olive gs-olive removed the WIP Work is in progress, pull request should not be merged yet label May 19, 2023
@gs-olive gs-olive changed the title examples: Add example usage scripts for torch_tensorrt.dynamo.dynamo_compile path examples: Add example usage scripts for torch_tensorrt.dynamo.compile path May 19, 2023
@gs-olive gs-olive force-pushed the dynamo_torch_compile_examples branch 2 times, most recently from 2f9019a to 4299036 Compare May 22, 2023 16:10
- Add sample scripts covering resnet18, transformers, and custom
examples showcasing the `torch_tensorrt.dynamo.compile` path,
which can compile models with data-dependent control flow and other such
restrictions which can make other compilation methods more difficult
- Cover different customizeable features allowed in the new backend
- Make scripts Sphinx-Gallery compatible Python files
@gs-olive gs-olive force-pushed the dynamo_torch_compile_examples branch from 4299036 to 63d2be8 Compare May 22, 2023 22:47
- Show individual links in sidebar
@gs-olive gs-olive closed this May 31, 2023
@gs-olive
Copy link
Contributor Author

Closed in favor of #1966

@gs-olive gs-olive deleted the dynamo_torch_compile_examples branch May 31, 2023 22:23
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

cla signed component: api [Python] Issues re: Python API component: dynamo Issues relating to the `torch.compile` or `torch._dynamo.export` paths documentation Improvements or additions to documentation

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants