common_defs.h

Go to the documentation of this file.

 
#ifndef VA_COMMON_DEFS_H
#define VA_COMMON_DEFS_H
 
// use 64-bit representation of real numbers
//#define VA_REAL double
// use 32-bit representation of real numbers
#define VA_REAL float
 
#define VA_PI 3.14159265358979323846
 
#if defined(CUDA)
    #include <cuda/std/complex>
    #define VA_DEVICE_ADDR
    #define VA_DEVICE_FUN __device__
    #define fabs fabsf
    #define sqrt sqrtf
    #define pow powf
    #define VA_COMPLEX cuda::std::complex<VA_REAL>
    #define VA_DOUBLE_COMPLEX cuda::std::complex<double>
#elif defined(HIP)
    #include <hip/hip_complex.h>
    #define VA_DEVICE_ADDR
    #define VA_DEVICE_FUN __device__
    #define fabs fabsf
    #define sqrt sqrtf
    #define pow powf
    // in HIP, it is difficult to distinguish float and double complex numbers
    #define VA_DOUBLE_COMPLEX hipDoubleComplex
    #define VA_COMPLEX VA_DOUBLE_COMPLEX
#else
    #include <math.h>
    #include <stdlib.h>
    #include <stdio.h>
    #include <complex.h>
    #include <float.h>
    #define VA_DEVICE_ADDR
    #define VA_DEVICE_FUN
    #define VA_COMPLEX complex VA_REAL
    #define VA_DOUBLE_COMPLEX complex double
    #define VA_INFINITY FLT_MAX
#endif
 
#endif // VA_COMMON_DEFS_H