Skip to content
This repository was archived by the owner on Mar 14, 2025. It is now read-only.

Commit 7cbd8a0

Browse files
committed
fix output for profile's first/last binding indices.
Signed-off-by: Ryan McCormick <rmccormick@nvidia.com>
1 parent 5816674 commit 7cbd8a0

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

inference/infer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -68,8 +68,8 @@ def get_binding_idxs(engine: trt.ICudaEngine, profile_index: int):
6868
print("Engine/Binding Metadata")
6969
print("\tNumber of optimization profiles: {}".format(engine.num_optimization_profiles))
7070
print("\tNumber of bindings per profile: {}".format(num_bindings_per_profile))
71-
print("\tFirst binding for profile: {}".format(profile_index))
72-
print("\tLast binding for profile: {}".format(profile_index))
71+
print("\tFirst binding for profile {}: {}".format(profile_index, start_binding))
72+
print("\tLast binding for profile {}: {}".format(profile_index, end_binding-1))
7373

7474
# Separate input and output binding indices for convenience
7575
input_binding_idxs = []

0 commit comments

Comments
 (0)