bug fix of operator "interp_linear"

752f29a1 · idontkonwher · GitHub · db4ae6e6 · 752f29a1
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddle/phi/kernels/gpu/interpolate_kernel.cu paddle/phi/kernels/gpu/interpolate_kernel.cu +1 -1

未找到文件。
--- a/paddle/phi/kernels/gpu/interpolate_kernel.cu
+++ b/paddle/phi/kernels/gpu/interpolate_kernel.cu
@@ -90,7 +90,7 @@ __global__ void KeLinearInterpFw(const T* in,

    if (data_layout == DataLayout::kNCHW) {
      const T* in_pos =
-          &in[out_id_h * out_id_w + channel_id * in_img_size + in_img_idx];
+          &in[out_id_h * input_w + channel_id * in_img_size + in_img_idx];
      // linear interpolation
      out[out_id_h * output_w + out_id_w] =
          static_cast<T>(w2lambda * static_cast<MT>(in_pos[0]) +