fix: Bugfix for align_corners=False- FX interpolate

gs-olive · gs-olive · commit 0c317ce5dc5a · 2022-12-19T11:07:44.000-08:00
- Improve argument checking and validation to match PyTorch interpolate
arguments
- Ensure `align_corners=None` and `align_corners=False` have the same
behavior
diff --git a/py/torch_tensorrt/fx/converters/acc_ops_converters.py b/py/torch_tensorrt/fx/converters/acc_ops_converters.py
@@ -3630,7 +3630,7 @@ def acc_ops_interpolate(
     else:
         layer.resize_mode = trt.ResizeMode.NEAREST
 
-    if align_corners != None:
+    if (align_corners is not None) and align_corners:
         layer.coordinate_transformation = (
             trt.ResizeCoordinateTransformation.ALIGN_CORNERS
         )

Original file line number	Diff line number	Diff line change
`@@ -3630,7 +3630,7 @@ def acc_ops_interpolate(`
`3630`	`3630`	`else:`
`3631`	`3631`	`layer.resize_mode = trt.ResizeMode.NEAREST`
`3632`	`3632`
`3633`		`- if align_corners != None:`
	`3633`	`+ if (align_corners is not None) and align_corners:`
`3634`	`3634`	`layer.coordinate_transformation = (`
`3635`	`3635`	`trt.ResizeCoordinateTransformation.ALIGN_CORNERS`
`3636`	`3636`	`)`