Skip to content

Commit b38fa5b

Browse files
authored
minor fix: Dynamo CI fix due to merge issue (#2067)
1 parent 66af8a4 commit b38fa5b

File tree

1 file changed

+3
-1
lines changed
  • py/torch_tensorrt/dynamo/fx_ts_compat

1 file changed

+3
-1
lines changed

py/torch_tensorrt/dynamo/fx_ts_compat/fx2trt.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -410,7 +410,9 @@ def output(self, target, args, kwargs):
410410
if output_bool:
411411
output.dtype = trt.bool
412412
elif self.output_dtypes is not None:
413-
output.dtype = torch_dtype_to_trt(self.output_dtypes[i])
413+
output.dtype = unified_dtype_converter(
414+
self.output_dtypes[i], Frameworks.TRT
415+
)
414416
elif self.output_fp16 and output.dtype == trt.float32:
415417
output.dtype = trt.float16
416418
self._output_names.append(name)

0 commit comments

Comments
 (0)