Fix upsample converter not properly registered

HolyWu · HolyWu · commit 5662ce1e727e · 2024-04-03T20:26:32.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
@@ -2509,6 +2509,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,
@@ -2530,6 +2531,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
`@@ -2509,6 +2509,7 @@ def aten_ops_pad(`
`2509`	`2509`	`)`
`2510`	`2510`
`2511`	`2511`
	`2512`	`+@dynamo_tensorrt_converter(torch.ops.aten.upsample_nearest2d.default)`
`2512`	`2513`	`@dynamo_tensorrt_converter(torch.ops.aten.upsample_nearest2d.vec)`
`2513`	`2514`	`def upsample_nearest2d(`
`2514`	`2515`	`ctx: ConversionContext,`
`@@ -2530,6 +2531,7 @@ def upsample_nearest2d(`
`2530`	`2531`	`)`
`2531`	`2532`
`2532`	`2533`
	`2534`	`+@dynamo_tensorrt_converter(torch.ops.aten.upsample_bilinear2d.default)`
`2533`	`2535`	`@dynamo_tensorrt_converter(torch.ops.aten.upsample_bilinear2d.vec)`
`2534`	`2536`	`def upsample_bilinear2d(`
`2535`	`2537`	`ctx: ConversionContext,`