Borrvall_Petersson_mapping_kernel.h

Go to the documentation of this file.

 
#ifndef __NEKO_HIP_BORRVALL_PETERSSON_MAPPING_KERNELS__
#define __NEKO_HIP_BORRVALL_PETERSSON_MAPPING_KERNELS__
 
template <typename T>
__global__ void Borrvall_Petersson_mapping_apply_kernel(
    const T f_min, const T f_max, const T q, T* __restrict__ X_out_d,
    T* __restrict__ X_in_d, const int n) {
 
    const int idx = blockIdx.x * blockDim.x + threadIdx.x;
    const int str = blockDim.x * gridDim.x;
 
    for (int i = idx; i < n; i += str) {
        X_out_d[i] = f_min
            + (f_max - f_min) * X_in_d[i] * (1.0 + q) / (X_in_d[i] + q);
    }
}
 
template <typename T>
__global__ void Borrvall_Petersson_mapping_apply_backward_kernel(
    const T f_min, const T f_max, const T q, T* __restrict__ sens_out_d,
    T* __restrict__ sens_in_d, T* __restrict__ X_in_d, const int n) {
 
    const int idx = blockIdx.x * blockDim.x + threadIdx.x;
    const int str = blockDim.x * gridDim.x;
 
    for (int i = idx; i < n; i += str) {
        sens_out_d[i] = (f_max - f_min) * q
            * (q + 1.0) / ( (X_in_d[i] + q) * (X_in_d[i] + q)) * sens_in_d[i];
    }
}
#endif // __NEKO_HIP_BORRVALL_PETERSSON_MAPPING_KERNELS__