-
Notifications
You must be signed in to change notification settings - Fork 370
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