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)