From dc26e0834dc8ebf4ad71067960fe3df7102e932d Mon Sep 17 00:00:00 2001 From: gs-olive <113141689+gs-olive@users.noreply.github.com> Date: Thu, 13 Jul 2023 17:47:35 -0700 Subject: [PATCH] fix: TRTInterpreter output lacks return value - Fixes error causing PyTree failures during Dynamo tracing for new converters returning lists of Tensors --- py/torch_tensorrt/dynamo/conversion/trt_interpreter.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/py/torch_tensorrt/dynamo/conversion/trt_interpreter.py b/py/torch_tensorrt/dynamo/conversion/trt_interpreter.py index 9f97fb1b0a..18fc050637 100644 --- a/py/torch_tensorrt/dynamo/conversion/trt_interpreter.py +++ b/py/torch_tensorrt/dynamo/conversion/trt_interpreter.py @@ -356,3 +356,5 @@ def output(self, target, args, kwargs): elif self.output_fp16 and output.dtype == trt.float32: output.dtype = trt.float16 self._output_names.append(name) + + return list(outputs)