#ifndef __MY_HEAD_H_
#define __MY_HEAD_H_

#ifdef USE_OPEN_MP
    #define OPEN_MP_THREAD_NUM 8
    #include "omp.h"
#endif

#ifdef _USE_CUDA_
    #include "cuda_runtime.h"
    #include "device_launch_parameters.h"
    #ifdef USE_SDK_BLAS
        #include "cublas_v2.h"
    #endif
#endif

#ifdef _USE_ROCM_
    #include <hip/hip_runtime.h>
    #ifdef USE_SDK_BLAS
        #include <hipblas/hipblas.h>
    #endif
#endif

#ifdef _USE_OPENCL_
#ifdef _USE_MAC
    #include "cl.h"
#else
    #include "CL/cl.h"
#endif
#endif

#endif
