File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 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 ,
You can’t perform that action at this time.
0 commit comments