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

/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-roccv/checkouts/latest/include/kernels/device/rotate_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/rotate_device.hpp Source File
rotate_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, typename MatWrapper>
28 __global__ void rotate(SrcWrapper src, DstWrapper dst, MatWrapper affineMat) {
29  const int x = blockDim.x * blockIdx.x + threadIdx.x;
30  const int y = blockDim.y * blockIdx.y + threadIdx.y;
31  const int b = blockIdx.z;
32 
33  if (x >= dst.width() || y >= dst.height()) return;
34 
35  const double xShift = x - affineMat[2];
36  const double yShift = y - affineMat[5];
37 
38  const float srcX = static_cast<float>(xShift * affineMat[0] + yShift * -affineMat[1]);
39  const float srcY = static_cast<float>(xShift * -affineMat[3] + yShift * affineMat[4]);
40 
41  dst.at(b, y, x, 0) = src.at(b, srcY, srcX, 0);
42 }
43 } // namespace Kernels::Device
Kernels::Device
Definition: bilateral_filter_device.hpp:35
Kernels::Device::rotate
__global__ void rotate(SrcWrapper src, DstWrapper dst, MatWrapper affineMat)
Definition: rotate_device.hpp:28