fix bug

zewenli98 · zewenli98 · commit 05e02c0fd2e4 · 2023-12-28T08:19:10.000-08:00
diff --git a/py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py b/py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py
@@ -2530,7 +2530,7 @@ def aten_ops_sort(
 @dynamo_tensorrt_converter(torch.ops.aten.trunc.default)
 @enforce_tensor_types(
     {
-        0: (TRTTensor,),
+        1: (TRTTensor,),
     }
 )
 def aten_ops_trunc(

Original file line number	Diff line number	Diff line change
`@@ -2530,7 +2530,7 @@ def aten_ops_sort(`
`2530`	`2530`	`@dynamo_tensorrt_converter(torch.ops.aten.trunc.default)`
`2531`	`2531`	`@enforce_tensor_types(`
`2532`	`2532`	`{`
`2533`		`- 0: (TRTTensor,),`
	`2533`	`+ 1: (TRTTensor,),`
`2534`	`2534`	`}`
`2535`	`2535`	`)`
`2536`	`2536`	`def aten_ops_trunc(`