Skip to content

Commit 95730fe

Browse files
authored
fix: TRTInterpreter output lacks return value (#2114)
1 parent 65277c5 commit 95730fe

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)