fix: Bugfix for align_corners=False- FX interpolate (#1561)

pytorch · Dec 20, 2022 · 0a075f0 · 0a075f0
1 parent f43be5b
commit 0a075f0
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
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
         )

diff --git a/py/torch_tensorrt/fx/test/converters/acc_op/test_interpolate.py b/py/torch_tensorrt/fx/test/converters/acc_op/test_interpolate.py
@@ -43,6 +43,14 @@ class TestInterpolateConverter(AccTestCase):
                 ("bilinear"),
                 (None),
             ),  # linear for 4D only
+            (
+                "4d_dim_scale_bilinear_align_corners_bool",
+                (2, 3, 4, 5),
+                (None),
+                (2),
+                ("bilinear"),
+                (False),
+            ),  # linear for 4D only
             (
                 "4d_dim_scale_align",
                 (2, 3, 4, 5),