Fix upsample converter not properly registered

HolyWu · HolyWu · commit d667f74957f6 · 2024-03-10T16:26:02.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
@@ -2491,6 +2491,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,
@@ -2512,6 +2513,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
`@@ -2491,6 +2491,7 @@ def aten_ops_pad(`
`2491`	`2491`	`)`
`2492`	`2492`
`2493`	`2493`
	`2494`	`+@dynamo_tensorrt_converter(torch.ops.aten.upsample_nearest2d.default)`
`2494`	`2495`	`@dynamo_tensorrt_converter(torch.ops.aten.upsample_nearest2d.vec)`
`2495`	`2496`	`def upsample_nearest2d(`
`2496`	`2497`	`ctx: ConversionContext,`
`@@ -2512,6 +2513,7 @@ def upsample_nearest2d(`
`2512`	`2513`	`)`
`2513`	`2514`
`2514`	`2515`
	`2516`	`+@dynamo_tensorrt_converter(torch.ops.aten.upsample_bilinear2d.default)`
`2515`	`2517`	`@dynamo_tensorrt_converter(torch.ops.aten.upsample_bilinear2d.vec)`
`2516`	`2518`	`def upsample_bilinear2d(`
`2517`	`2519`	`ctx: ConversionContext,`