Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

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

Closed
wants to merge 2 commits into from
Closed

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

wants to merge 2 commits into from

Conversation

gs-olive
Copy link
Collaborator

@gs-olive gs-olive commented May 5, 2023 •
edited
Loading

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
@gs-olive
feat: Add example usage scripts for dynamo path
63d2be8 
- 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
@gs-olive
fix: Update index.rst
cd6542e 
- Show individual links in sidebar
@gs-olive gs-olive closed this May 31, 2023
@gs-olive
Copy link
Collaborator 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
Reviewers

@narendasan narendasan Awaiting requested review from narendasan

Assignees

@gs-olive gs-olive

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
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@gs-olive @facebook-github-bot