From 169988611390f4544ce3ab09f84b9150c3787ae5 Mon Sep 17 00:00:00 2001 From: gs-olive <113141689+gs-olive@users.noreply.github.com> Date: Wed, 28 Jun 2023 15:56:18 -0700 Subject: [PATCH] fix: Dynamo CI issue due to minor typo --- py/torch_tensorrt/dynamo/fx_ts_compat/fx2trt.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/py/torch_tensorrt/dynamo/fx_ts_compat/fx2trt.py b/py/torch_tensorrt/dynamo/fx_ts_compat/fx2trt.py index eb9924f2b9..a29cee509d 100644 --- a/py/torch_tensorrt/dynamo/fx_ts_compat/fx2trt.py +++ b/py/torch_tensorrt/dynamo/fx_ts_compat/fx2trt.py @@ -410,7 +410,9 @@ def output(self, target, args, kwargs): if output_bool: output.dtype = trt.bool elif self.output_dtypes is not None: - output.dtype = torch_dtype_to_trt(self.output_dtypes[i]) + output.dtype = unified_dtype_converter( + self.output_dtypes[i], Frameworks.TRT + ) elif self.output_fp16 and output.dtype == trt.float32: output.dtype = trt.float16 self._output_names.append(name)