Skip to content

📖 [Story] Improve the torch_tensorrt.dynamo.compile path #1941

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.

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

Closed
gs-olive opened this issue May 22, 2023 · 1 comment
Closed

📖 [Story] Improve the torch_tensorrt.dynamo.compile path #1941

gs-olive opened this issue May 22, 2023 · 1 comment
Assignees
Labels
Story: Dynamo Compile Improvements Issues relating to improvement of the Dynamo compile path Story Issues proposing a new Story

Comments

@gs-olive
Copy link
Collaborator

TL;DR

Improve the performance acceleration, robustness, and overall quality of the torch_tensorrt.dynamo.compile path. Simultaneously use these improvements to improve the quality of the torch_tensorrt.dynamo.fx_ts_compat path.

Goal(s)

By adding lowering passes, improving converter coverage, and building features to accelerate the performance of our torch.compile backend, we can improve the quality of the path. Additionally, in conjunction with #1940, many of these upgrades can be ported over to our torch._dynamo.export path.

Implementation Phases

  1. Module-Level Acceleration ✨[Feature] Upgrade Prototype System for Module-Level Acceleration in Dynamo Path #1894
  2. Improved Converter Coverage (↔ [Converter] Add support for assorted operators in the FX aten path #1769, ↔ [Converter] Add operations to accelerate Transformer encoder #1754)
  3. Support for truncate_long_and_double (fix: Add support for truncate_long_and_double in FX #1865)
  4. Engine caching + cross-session state capture
  5. Dynamic shape support + avoiding unnecessary recompilation

Additional context

Related to #1940

@gs-olive gs-olive added the Story Issues proposing a new Story label May 22, 2023
@gs-olive gs-olive self-assigned this May 22, 2023
@gs-olive gs-olive added Story: Dynamo Compile Improvements Issues relating to improvement of the Dynamo compile path and removed Story Issues proposing a new Story labels May 22, 2023
@narendasan narendasan added the Story Issues proposing a new Story label Jun 21, 2023
@github-actions
Copy link

This issue has not seen activity for 90 days, Remove stale label or comment or this will be closed in 10 days

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Story: Dynamo Compile Improvements Issues relating to improvement of the Dynamo compile path Story Issues proposing a new Story
Projects
None yet
Development

No branches or pull requests

2 participants