fix: TRTInterpreter output lacks return value (#2114)

pytorch · Jul 21, 2023 · 95730fe · 95730fe
1 parent 65277c5
commit 95730fe
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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)