Skip to content

Commit 7f9a4a6

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 9cb57c2 commit 7f9a4a6

File tree

1 file changed

+2
-0
lines changed
  • py/torch_tensorrt/dynamo/fx_ts_compat

1 file changed

+2
-0
lines changed

py/torch_tensorrt/dynamo/fx_ts_compat/fx2trt.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -416,3 +416,5 @@ def output(self, target, args, kwargs):
416416
elif self.output_fp16 and output.dtype == trt.float32:
417417
output.dtype = trt.float16
418418
self._output_names.append(name)
419+
420+
return list(outputs)

0 commit comments

Comments
 (0)