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

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

2 min read time

Applies to Linux

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

Go to the documentation of this file.

  
 #pragma once
  
 #include <hip/hip_runtime.h>
  
 #include "operator_types.h"
  
 namespace Kernels {
 namespace Device {
 template <typename SrcWrapper, typename DstWrapper>
 __global__ void custom_crop(SrcWrapper input, DstWrapper output, Box_t cropRect) {
     const int x = blockIdx.x * blockDim.x + threadIdx.x;
     const int y = blockIdx.y * blockDim.y + threadIdx.y;
     const int b = blockIdx.z;
  
     if (x >= cropRect.width || y >= cropRect.height || b >= output.batches()) return;
  
     const int srcX = x + cropRect.x;
     const int srcY = y + cropRect.y;
     const int dstX = x;
     const int dstY = y;
  
     output.at(b, dstY, dstX, 0) = input.at(b, srcY, srcX, 0);
 }
 }  // namespace Device
 }  // namespace Kernels