Fix upsample converter not properly registered (#2683)

HolyWu · peri044 · commit 6ea06d91bea0 · 2024-04-18T21:14:04.000-07:00
diff --git a/py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py b/py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py
@@ -2639,6 +2639,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,
@@ -2660,6 +2661,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
`@@ -2639,6 +2639,7 @@ def aten_ops_pad(`
`2639`	`2639`	`)`
`2640`	`2640`
`2641`	`2641`
	`2642`	`+@dynamo_tensorrt_converter(torch.ops.aten.upsample_nearest2d.default)`
`2642`	`2643`	`@dynamo_tensorrt_converter(torch.ops.aten.upsample_nearest2d.vec)`
`2643`	`2644`	`def upsample_nearest2d(`
`2644`	`2645`	`ctx: ConversionContext,`
`@@ -2660,6 +2661,7 @@ def upsample_nearest2d(`
`2660`	`2661`	`)`
`2661`	`2662`
`2662`	`2663`
	`2664`	`+@dynamo_tensorrt_converter(torch.ops.aten.upsample_bilinear2d.default)`
`2663`	`2665`	`@dynamo_tensorrt_converter(torch.ops.aten.upsample_bilinear2d.vec)`
`2664`	`2666`	`def upsample_bilinear2d(`
`2665`	`2667`	`ctx: ConversionContext,`