CudaFunctions.h

#ifndef _CUDAFUNCTIONS_H_
#define _CUDAFUNCTIONS_H_
 
#ifdef CUDA
#include "hapi.h"
#include "HostCUDA.h"
void TreePieceDataTransferBasic(CudaRequest *data, CudaDevPtr *ptr);
void TreePieceDataTransferBasicCleanup(CudaDevPtr *ptr);
 
#ifdef GPU_LOCAL_TREE_WALK
__global__ void gpuLocalTreeWalk(
    CudaMultipoleMoments* moments,
    CompactPartData *particleCores,
    VariablePartData *particleVars,
    int firstParticle,
    int lastParticle,
    int rootIdx,
    cudatype theta,
    cudatype thetaMono,
    int nReplicas,
    cudatype fperiod,
    cudatype fperiodY,
    cudatype fperiodZ);
#endif //GPU_LOCAL_TREE_WALK
 
__global__ void nodeGravityComputation(
        CompactPartData *particleCores,
        VariablePartData *particleVars,
        CudaMultipoleMoments *moments,
        ILCell *ils,
        int *ilmarks,
        int *bucketStarts,
        int *bucketSizes,
        cudatype fperiod
                );
 
__global__ void ZeroVars(VariablePartData *particleVars, int nVars);
 
#ifdef CUDA_2D_TB_KERNEL
__global__ void particleGravityComputation(
                CompactPartData *targetCores,
                VariablePartData *targetVars,
                CompactPartData *sourceCores,
                ILCell *ils,
                int *ilmarks,
        int *bucketStarts,
        int *bucketSizes,
        cudatype fperiod);
#else
__global__ void particleGravityComputation(
                CompactPartData *targetCores,
                VariablePartData *targetVars,
                CompactPartData *sourceCores,
                ILPart *ils,
                int *ilmarks,
        int *bucketStarts,
        int *bucketSizes,
        cudatype fperiod);
#endif
 
#endif
 
#endif