Skip to content

Commit cc706de

Browse files
scripts/vsmlrt.py: cosmetics
1 parent 7a52950 commit cc706de

1 file changed

Lines changed: 2 additions & 5 deletions

File tree

scripts/vsmlrt.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
__version__ = "3.22.32"
1+
__version__ = "3.22.33"
22

33
__all__ = [
44
"Backend", "BackendV2",
@@ -1880,7 +1880,6 @@ def get_engine_path(
18801880
max_shapes: typing.Tuple[int, int],
18811881
workspace: typing.Optional[int],
18821882
fp16: bool,
1883-
device_id: int,
18841883
use_cublas: bool,
18851884
static_shape: bool,
18861885
tf32: bool,
@@ -2014,7 +2013,6 @@ def trtexec(
20142013
max_shapes=max_shapes,
20152014
workspace=workspace,
20162015
fp16=fp16,
2017-
device_id=device_id,
20182016
use_cublas=use_cublas,
20192017
static_shape=static_shape,
20202018
tf32=tf32,
@@ -2416,11 +2414,10 @@ def tensorrt_rtx(
24162414
max_shapes=max_shapes,
24172415
workspace=workspace,
24182416
fp16=fp16,
2419-
device_id=device_id,
24202417
use_cublas=False,
24212418
static_shape=static_shape,
24222419
tf32=False,
2423-
use_cudnn=False,
2420+
use_cudnn=use_cudnn,
24242421
input_format=int(fp16_io),
24252422
output_format=int(fp16_io),
24262423
builder_optimization_level=builder_optimization_level,

0 commit comments

Comments
 (0)