/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-roccv/checkouts/latest/include/kernels/device/resize_device.hpp Source File

/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-roccv/checkouts/latest/include/kernels/device/resize_device.hpp Source File#

3 min read time

Applies to Linux

rocCV: /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-roccv/checkouts/latest/include/kernels/device/resize_device.hpp Source File
resize_device.hpp
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2025 Advanced Micro Devices, Inc. All rights reserved.
3  * Permission is hereby granted, free of charge, to any person obtaining a copy
4  * of this software and associated documentation files (the "Software"), to deal
5  * in the Software without restriction, including without limitation the rights
6  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
7  * copies of the Software, and to permit persons to whom the Software is
8  * furnished to do so, subject to the following conditions:
9  *
10  * The above copyright notice and this permission notice shall be included in
11  * all copies or substantial portions of the Software.
12  *
13  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
14  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
15  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
16  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
17  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
18  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
19  * THE SOFTWARE.
20  */
21 
22 #pragma once
23 
24 #include <hip/hip_runtime.h>
25 
26 namespace Kernels::Device {
27 template <typename SrcWrapper, typename DstWrapper>
28 __global__ void resize(SrcWrapper input, DstWrapper output, float scaleX, float scaleY) {
29  const int x = blockIdx.x * blockDim.x + threadIdx.x;
30  const int y = blockIdx.y * blockDim.y + threadIdx.y;
31  const int batch = blockIdx.z;
32 
33  if (x >= output.width() || y >= output.height()) return;
34 
35  float srcX = fmaf(x + 0.5f, scaleX, -0.5f);
36  float srcY = fmaf(y + 0.5f, scaleY, -0.5f);
37  output.at(batch, y, x, 0) = input.at(batch, srcY, srcX, 0);
38 }
39 } // namespace Kernels::Device
Kernels::Device
Definition: bilateral_filter_device.hpp:35
Kernels::Device::resize
__global__ void resize(SrcWrapper input, DstWrapper output, float scaleX, float scaleY)
Definition: resize_device.hpp:28