Skip to content

Commit 0a075f0

Browse files
authored
fix: Bugfix for align_corners=False- FX interpolate (#1561)
1 parent f43be5b commit 0a075f0

File tree

2 files changed

+9
-1
lines changed

2 files changed

+9
-1
lines changed

py/torch_tensorrt/fx/converters/acc_ops_converters.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3630,7 +3630,7 @@ def acc_ops_interpolate(
36303630
else:
36313631
layer.resize_mode = trt.ResizeMode.NEAREST
36323632

3633-
if align_corners != None:
3633+
if (align_corners is not None) and align_corners:
36343634
layer.coordinate_transformation = (
36353635
trt.ResizeCoordinateTransformation.ALIGN_CORNERS
36363636
)

py/torch_tensorrt/fx/test/converters/acc_op/test_interpolate.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,14 @@ class TestInterpolateConverter(AccTestCase):
4343
("bilinear"),
4444
(None),
4545
), # linear for 4D only
46+
(
47+
"4d_dim_scale_bilinear_align_corners_bool",
48+
(2, 3, 4, 5),
49+
(None),
50+
(2),
51+
("bilinear"),
52+
(False),
53+
), # linear for 4D only
4654
(
4755
"4d_dim_scale_align",
4856
(2, 3, 4, 5),

0 commit comments

Comments
 (0)