pytorch / pytorch/TensorRT

✨[Feature] Explore using the STABLE_TORCH_LIBRARY API

Open
#3,914 0 comments 0 reactions 1 assignee View on GitHub

Nobody has claimed this yet.

feature request story: Dynamo Frontend & Partitioning
Dominant language
Python
Stars
3k
Forks
410
Avg merge
3d 18h
Merged PRs (30d)
78

Description

Is your feature request related to a problem? Please describe.

Torch-TensorRT builds are deeply tied to the PyTorch version they are compiled with. Typically a lot of import errors are due to version mismatch issue therefore to use latest Torch-TensorRT users are forced to upgrade their Torch version

Describe the solution you'd like

Use the STABLE_TORCH_LIBRARY API https://docs.pytorch.org/docs/stable/notes/libtorch_stable_abi.html to solve these ABI incompatibility issues

Describe alternatives you've considered

Python-only solve this problem as well but has significant drawbacks when it comes to workflow support

Additional context

While this feature is developing we should probably conditionally build it so maybe maintain a stable and non stable registration that we control with a bazel flag

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.