Fix upsample converter not properly registered (#2683)

HolyWu · web-flow · commit dde535e3e1da · 2024-04-18T19:06:27.000-06:00
diff --git a/py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py b/py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py
@@ -2623,6 +2623,7 @@ def aten_ops_pad(
     )
 
 
+@dynamo_tensorrt_converter(torch.ops.aten.upsample_nearest2d.default)
 @dynamo_tensorrt_converter(torch.ops.aten.upsample_nearest2d.vec)
 def upsample_nearest2d(
     ctx: ConversionContext,
@@ -2644,6 +2645,7 @@ def upsample_nearest2d(
     )
 
 
+@dynamo_tensorrt_converter(torch.ops.aten.upsample_bilinear2d.default)
 @dynamo_tensorrt_converter(torch.ops.aten.upsample_bilinear2d.vec)
 def upsample_bilinear2d(
     ctx: ConversionContext,

Original file line number	Diff line number	Diff line change
`@@ -2623,6 +2623,7 @@ def aten_ops_pad(`
`2623`	`2623`	`)`
`2624`	`2624`
`2625`	`2625`
	`2626`	`+@dynamo_tensorrt_converter(torch.ops.aten.upsample_nearest2d.default)`
`2626`	`2627`	`@dynamo_tensorrt_converter(torch.ops.aten.upsample_nearest2d.vec)`
`2627`	`2628`	`def upsample_nearest2d(`
`2628`	`2629`	`ctx: ConversionContext,`
`@@ -2644,6 +2645,7 @@ def upsample_nearest2d(`
`2644`	`2645`	`)`
`2645`	`2646`
`2646`	`2647`
	`2648`	`+@dynamo_tensorrt_converter(torch.ops.aten.upsample_bilinear2d.default)`
`2647`	`2649`	`@dynamo_tensorrt_converter(torch.ops.aten.upsample_bilinear2d.vec)`
`2648`	`2650`	`def upsample_bilinear2d(`
`2649`	`2651`	`ctx: ConversionContext,`