Skip to content

Commit

Permalink
[TensorRT EP] update gen_trt_engine_wrapper_onnx_model.py script (#22184
Browse files Browse the repository at this point in the history
)

update script which was using deprecated num_bindings to num_io_tensors
tested on an engine dumped by trtexec and loaded the engine using
onnxruntime-gpu 1.19.2 python package.
  • Loading branch information
jywu-msft authored Sep 24, 2024
1 parent 6cc06ad commit 7727b4b
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@ def __init__(self, args):
# Deserialize an TRT engine
runtime = trt.Runtime(logger)
engine = runtime.deserialize_cuda_engine(engine_buffer)
num_bindings = engine.num_bindings
num_bindings = engine.num_io_tensors

input_tensors = []
output_tensors = []
Expand Down

0 comments on commit 7727b4b

Please sign in to comment.