From adf3415d2d258d8b43f2a2317f9855d6d76cbe36 Mon Sep 17 00:00:00 2001
From: Naren Dasan <1790613+narendasan@users.noreply.github.com>
Date: Thu, 30 Jan 2025 16:22:00 -0700
Subject: [PATCH] chore: moving away from tensorrt_bindings (#3365)

---
 dev_dep_versions.yml                                     | 2 +-
 examples/dynamo/auto_generate_converters.py              | 2 +-
 .../conversion/plugins/_generate_plugin_converter.py     | 4 ++--
 pyproject.toml                                           | 9 +++++----
 4 files changed, 9 insertions(+), 8 deletions(-)

diff --git a/dev_dep_versions.yml b/dev_dep_versions.yml
index b921d558fd..6ba35b5ef9 100644
--- a/dev_dep_versions.yml
+++ b/dev_dep_versions.yml
@@ -1,2 +1,2 @@
 __cuda_version__: "12.6"
-__tensorrt_version__: "10.7.0"
+__tensorrt_version__: "10.7.0.post1"
diff --git a/examples/dynamo/auto_generate_converters.py b/examples/dynamo/auto_generate_converters.py
index 0640f4f104..59085fe2fb 100644
--- a/examples/dynamo/auto_generate_converters.py
+++ b/examples/dynamo/auto_generate_converters.py
@@ -39,7 +39,7 @@
 
 from typing import Tuple
 
-import tensorrt_bindings.plugin as trtp
+import tensorrt.plugin as trtp
 import torch
 import torch_tensorrt
 import triton
diff --git a/py/torch_tensorrt/dynamo/conversion/plugins/_generate_plugin_converter.py b/py/torch_tensorrt/dynamo/conversion/plugins/_generate_plugin_converter.py
index 48886b82df..b880939b17 100644
--- a/py/torch_tensorrt/dynamo/conversion/plugins/_generate_plugin_converter.py
+++ b/py/torch_tensorrt/dynamo/conversion/plugins/_generate_plugin_converter.py
@@ -4,9 +4,9 @@
 import numpy as np
 
 # Seems like a bug in TensorRT
-import tensorrt_bindings.plugin as trtp
+import tensorrt.plugin as trtp
 import torch
-from tensorrt_bindings.plugin._lib import QDP_REGISTRY
+from tensorrt.plugin._lib import QDP_REGISTRY
 from torch.fx.node import Argument, Node, Target
 from torch_tensorrt.dynamo._settings import CompilationSettings
 from torch_tensorrt.dynamo.conversion._ConversionContext import ConversionContext
diff --git a/pyproject.toml b/pyproject.toml
index 0472b39f12..83fe1ff831 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -8,7 +8,7 @@ requires = [
     "cffi>=1.15.1",
     "typing-extensions>=4.7.0",
     "future>=0.18.3",
-    "tensorrt-cu12>=10.6.0,<10.8.0",
+    "tensorrt-cu12>=10.7.0.post1,<10.8.0",
     "torch>=2.6.0.dev,<2.7.0",
     "pybind11==2.6.2",
     "numpy",
@@ -55,9 +55,10 @@ keywords = [
 ]
 dependencies = [
     "torch>=2.6.0.dev,<2.7.0",
-    "tensorrt-cu12>=10.6.0,<10.8.0",
-    "tensorrt-cu12-bindings>=10.6.0,<10.8.0",
-    "tensorrt-cu12-libs>=10.6.0,<10.8.0",
+    "tensorrt>=10.7.0.post1,<10.8.0",
+    "tensorrt-cu12>=10.7.0.post1,<10.8.0",
+    "tensorrt-cu12-bindings>=10.7.0,<10.8.0",
+    "tensorrt-cu12-libs>=10.7.0,<10.8.0",
     "packaging>=23",
     "numpy",
     "typing-extensions>=4.7.0",