diff --git a/perception/autoware_tensorrt_plugins/src/bev_ops/bev_pool_cuda.cu b/perception/autoware_tensorrt_plugins/src/bev_ops/bev_pool_cuda.cu index baf1654c76222..906a26d760863 100644 --- a/perception/autoware_tensorrt_plugins/src/bev_ops/bev_pool_cuda.cu +++ b/perception/autoware_tensorrt_plugins/src/bev_ops/bev_pool_cuda.cu @@ -32,7 +32,7 @@ __global__ void bev_pool_kernel( int interval_start = interval_starts[index]; int interval_length = interval_lengths[index]; - const int * cur_geom_feats = geom_feats + interval_start * 4; + const int * cur_geom_feats = geom_feats + (interval_start + interval_length -1) * 4; const float * cur_x = x + interval_start * c + cur_c; float * cur_out = out + cur_geom_feats[3] * d * h * w * c + cur_geom_feats[2] * h * w * c + cur_geom_feats[0] * w * c + cur_geom_feats[1] * c + cur_c;