Skip to content

Commit a4fb224

Browse files
authored
chore: moving away from tensorrt_bindings (#3365)
1 parent 002bacf commit a4fb224

File tree

4 files changed

+7
-6
lines changed

4 files changed

+7
-6
lines changed

dev_dep_versions.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,2 @@
11
__cuda_version__: "12.6"
2-
__tensorrt_version__: "10.7.0"
2+
__tensorrt_version__: "10.7.0.post1"

examples/dynamo/auto_generate_converters.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@
3939

4040
from typing import Tuple
4141

42-
import tensorrt_bindings.plugin as trtp
42+
import tensorrt.plugin as trtp
4343
import torch
4444
import torch_tensorrt
4545
import triton

py/torch_tensorrt/dynamo/conversion/plugins/_generate_plugin_converter.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -4,9 +4,9 @@
44
import numpy as np
55

66
# Seems like a bug in TensorRT
7-
import tensorrt_bindings.plugin as trtp
7+
import tensorrt.plugin as trtp
88
import torch
9-
from tensorrt_bindings.plugin._lib import QDP_REGISTRY
9+
from tensorrt.plugin._lib import QDP_REGISTRY
1010
from torch.fx.node import Argument, Node, Target
1111
from torch_tensorrt.dynamo._settings import CompilationSettings
1212
from torch_tensorrt.dynamo.conversion._ConversionContext import ConversionContext

pyproject.toml

+3-2
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ requires = [
88
"cffi>=1.15.1",
99
"typing-extensions>=4.7.0",
1010
"future>=0.18.3",
11-
"tensorrt-cu12>=10.7.0,<10.8.0",
11+
"tensorrt-cu12>=10.7.0.post1,<10.8.0",
1212
"torch>=2.7.0.dev,<2.8.0",
1313
"pybind11==2.6.2",
1414
"numpy",
@@ -55,7 +55,8 @@ keywords = [
5555
]
5656
dependencies = [
5757
"torch>=2.7.0.dev,<2.8.0",
58-
"tensorrt-cu12>=10.7.0,<10.8.0",
58+
"tensorrt>=10.7.0.post1,<10.8.0",
59+
"tensorrt-cu12>=10.7.0.post1,<10.8.0",
5960
"tensorrt-cu12-bindings>=10.7.0,<10.8.0",
6061
"tensorrt-cu12-libs>=10.7.0,<10.8.0",
6162
"packaging>=23",

0 commit comments

Comments
 (0)