Skip to content

Commit dc26e08

Browse files
committed
fix: TRTInterpreter output lacks return value
- Fixes error causing PyTree failures during Dynamo tracing for new converters returning lists of Tensors
1 parent 65277c5 commit dc26e08

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

py/torch_tensorrt/dynamo/conversion/trt_interpreter.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -356,3 +356,5 @@ def output(self, target, args, kwargs):
356356
elif self.output_fp16 and output.dtype == trt.float32:
357357
output.dtype = trt.float16
358358
self._output_names.append(name)
359+
360+
return list(outputs)

0 commit comments

Comments
 (0)