Skip to content

Commit ab4d85e

Browse files
committed
Fix issue microsoft#10607, grid_sample CUDA kernel clamping is incorrect for border padding mode with align = 1.
1 parent eeb7007 commit ab4d85e

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

onnxruntime/contrib_ops/cuda/grid_sample_impl.cu

+3-2
Original file line numberDiff line numberDiff line change
@@ -167,8 +167,9 @@ __global__ void _GridSampleKernel(
167167
if (grid_x_imgSpace < x_min || grid_x_imgSpace > x_max ||
168168
grid_y_imgSpace < y_min || grid_y_imgSpace > y_max) { // out of bound
169169
if (padding_mode == 1) { // border
170-
grid_x_imgSpace = max(0.0f, min(grid_x_imgSpace, W_in - 1.0f));
171-
grid_y_imgSpace = max(0.0f, min(grid_y_imgSpace, H_in - 1.0f));
170+
// Clamping must not be done here, see #10607
171+
//grid_x_imgSpace = max(0.0f, min(grid_x_imgSpace, W_in - 1.0f));
172+
//grid_y_imgSpace = max(0.0f, min(grid_y_imgSpace, H_in - 1.0f));
172173
} else if (padding_mode == 2) { // reflection
173174
grid_x_imgSpace = GsReflect(grid_x_imgSpace, x_min, x_max);
174175
grid_y_imgSpace = GsReflect(grid_y_imgSpace, y_min, y_max);

0 commit comments

Comments
 (0)