Skip to content
Snippets Groups Projects
Select Git revision
  • 732d1d449c4c3b8628c57b1e19913866a1e76aaa
  • main default protected
  • merge
  • split_input
  • revamp
  • test_derubeis
  • glacopo
  • inverse_imaging
  • openmp
  • RICKv2.0
  • RICKv1.0
  • r0.1c
  • end-of-re_structuring
13 results

w-stacking.hip.hpp

Blame
  • w-stacking.hip.hpp 1.34 KiB
    #ifndef W_PROJECT_H_
    #define W_PROJECT_H_
    
    #define NWORKERS -1    //100
    #define NTHREADS 64    //AMD WARP SIZE
    #define PI 3.14159265359
    #define REAL_TYPE double
    
    #include <mpi.h>
    #include <hip/hip_runtime.h>
    
    #ifdef __HIPCC__
    extern "C"
    #endif
    
    #ifdef __cplusplus
    extern "C" {
    void wstack(
         int,
         unsigned int,
         unsigned int,
         unsigned int,
         double*,
         double*,
         double*,
         float*,
         float*,
         float*,
         double,
         double,
         int,
         int,
         int,
         double*,
         int,
         int,
         hipStream_t);
    }
    
    #else 
    void wstack(
         int,
         unsigned int,
         unsigned int,
         unsigned int,
         double*,
         double*,
         double*,
         float*,
         float*,
         float*,
         double,
         double,
         int,
         int,
         int,
         double*,
         int,
         int);
    #endif
    
    
    
    #ifdef __HIPCC__
    extern "C"
    #endif
    int test(int nnn);
    
    #ifdef __HIPCC__
    extern "C"
    #endif
    void phase_correction(
         double*,
         double*,
         double*,
         int,
         int,
         int,
         int,
         int,
         double,
         double,
         double,
         int,
         int);
    
    double gauss_kernel_norm(
      double norm,
      double std22,
      double u_dist,
      double v_dist);
    
    #ifdef ACCOMP
    #pragma omp declare target (gauss_kernel_norm)
    #endif
    
    #ifdef __HIPCC__
    extern "C"
    #endif
    
    void cuda_fft(
    	int,
    	int,
    	int,
    	int,
    	int,
    	double*,
    	double*,
    	int,
    	MPI_Comm);
    
    
    #endif