cuda_util.hpp

/*
 * cuda_util.hpp
 *
 *  Created on: Jun 13, 2018
 *      Author: i-bird
 */
 
#ifndef OPENFPM_DATA_SRC_UTIL_CUDA_UTIL_HPP_
#define OPENFPM_DATA_SRC_UTIL_CUDA_UTIL_HPP_
 
#include "config.h"
 
#if defined(CUDIFY_USE_ALPAKA)
#define CUDA_ON_CPU
#elif defined(CUDIFY_USE_OPENMP)
#define CUDA_ON_CPU
#elif defined(CUDIFY_USE_SEQUENTIAL)
#define CUDA_ON_CPU
#endif
 
#if defined(__HIP__)
    // If hip fill NVCC think it is Nvidia compiler
    #ifdef __NVCC__
        #undef __NVCC__
        #include <hip/hip_runtime.h>
        #define __NVCC__
    #else
        #include <hip/hip_runtime.h>
    #endif
#elif defined(CUDIFY_USE_CUDA)
#include <cuda_runtime.h>
#endif
 
#ifdef CUDA_GPU
 
    #ifndef __NVCC__
 
        #ifndef __host__
        #define __host__
        #define __device__
        #define __shared__ static thread_local
        #define __global__ inline
        #endif
 
    #else
 
        #ifndef __host__
        #define __host__
        #define __device__
        #define __global__ inline
        #endif
 
        #ifdef CUDA_ON_CPU 
            
            #ifdef __shared__
                #undef __shared__
            #endif
            #define __shared__ static thread_local
 
        #else
 
            #ifndef __shared__
            #define __shared__
            #endif
 
        #endif
 
    #endif
#else
 
#ifndef __host__
#define __host__
#define __device__
#define __shared__ static thread_local
#define __global__ inline
#endif
 
#endif
 
 
#endif /* OPENFPM_DATA_SRC_UTIL_CUDA_UTIL_HPP_ */