Skip to content

fix(vad-trt): fix TensorRT input ordering using engine tensor names#127

Open
SZucchini wants to merge 1 commit into
NVIDIA:masterfrom
SZucchini:SZucchini/fix-vad-trt-input-order
Open

fix(vad-trt): fix TensorRT input ordering using engine tensor names#127
SZucchini wants to merge 1 commit into
NVIDIA:masterfrom
SZucchini:SZucchini/fix-vad-trt-input-order

Conversation

@SZucchini
Copy link
Copy Markdown

What

Replace hardcoded tensor pointer ordering with a name-keyed tensor_map, using infer.names to derive the correct argument order at runtime.

Why

AutoInspectHelper.export() applies g.toposort() and onnxsim.simplify() during ONNX export, which can reorder graph inputs. The hardcoded order did not reliably match the compiled engine's input order, causing incorrect inference outputs.

Signed-off-by: SZucchini <tomohiro.suzuki@hdwlab.co.jp>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant