Skip to content

Commit 08a13e4

Browse files
peri044narendasan
authored andcommitted
chore: Fix interpolate align corners deprecation
Signed-off-by: Dheeraj Peri <[email protected]>
1 parent 0dd5e23 commit 08a13e4

File tree

1 file changed

+2
-10
lines changed

1 file changed

+2
-10
lines changed

core/conversion/converters/impl/interpolate.cpp

Lines changed: 2 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -109,17 +109,9 @@ void resize_layer_size(
109109
resize_layer->setResizeMode(mode);
110110
resize_layer->setName(util::node_info(n).c_str());
111111

112-
// if interpolation mode is linear, align corners must have been set to true.
113-
// else, don't use align corners.
114-
if (mode == nvinfer1::ResizeMode::kLINEAR) {
115-
#if NV_TENSORRT_MAJOR < 7 || (NV_TENSORRT_MAJOR == 7 && NV_TENSORRT_MINOR < 1) // IF TRT VERSION <= 7.0
116-
TRTORCH_CHECK(align_corners, "resize layer (linear) only supports align_corners=True in TensorRT <= 7.0");
117-
resize_layer->setAlignCorners(true);
118-
#else
119-
resize_layer->setAlignCorners(align_corners);
120-
#endif
112+
if (align_corners) {
113+
resize_layer->setCoordinateTransformation(nvinfer1::ResizeCoordinateTransformation::kALIGN_CORNERS);
121114
}
122-
123115
auto layer_output = ctx->AssociateValueAndTensor(n->outputs()[0], resize_layer->getOutput(0));
124116

125117
LOG_DEBUG("Output tensor shape: " << layer_output->getDimensions());

0 commit comments

Comments
 (0)