heaviside_mapping.cu

Go to the documentation of this file.

 
// System includes
#include <stdio.h>
#include <stdlib.h>
 
// Device includes
#include <cuda_runtime.h>
 
// Neko includes
#include <neko/device/cuda/check.h>
#include <neko/device/device_config.h>
 
// Local includes
#include "heaviside_mapping_kernel.h"
 
extern "C" {
 
void cuda_heaviside_mapping_apply(real* beta, real* eta,
    void* X_out_d, void* X_in_d, int* n) {
 
    const dim3 nthrds(1024, 1, 1);
    const dim3 nblcks(((*n) + 1024 - 1) / 1024, 1, 1);
 
    heaviside_mapping_apply_kernel<real>
        <<<nblcks, nthrds, 0, (cudaStream_t)glb_cmd_queue>>>
        (*beta, *eta, (real*)X_out_d, (real*)X_in_d, *n);
    CUDA_CHECK(cudaGetLastError());
}
 
void cuda_heaviside_mapping_apply_backward(real* beta, real* eta,
    void* sens_out_d, void* sens_in_d, void* X_in_d, int* n) {
 
    const dim3 nthrds(1024, 1, 1);
    const dim3 nblcks(((*n) + 1024 - 1) / 1024, 1, 1);
 
    heaviside_mapping_apply_backward_kernel<real>
        <<<nblcks, nthrds, 0, (cudaStream_t)glb_cmd_queue>>>
        (*beta, *eta, (real*)sens_out_d, (real*)sens_in_d, (real*)X_in_d, *n);
    CUDA_CHECK(cudaGetLastError());
}
 
}