Feed every input a model declares when probing a rung

Browse Source

The probe built one zero tensor from the first input and ran the session
with it. Every model so far had one input; the denoiser has two (mosaic and
sigma), so every rung failed with "Missing Input: sigma" and the role was
left on the CPU: 14.4 s for a 20 MP frame where TensorRT fp16 takes 3.1 s.
Zeros now go to each input by name.

...

This commit is contained in:

dtourolle

2026-10-03 11:15:49 -04:00

parent 6b0d29cc15

commit 20b7bd7663

2 changed files with 27 additions and 15 deletions

Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL

Download Patch File Download Diff File Expand all files Collapse all files

docs/dev/traceability.md

+3 -3

File diff suppressed because one or more lines are too long