CudaGlobalMasterServerKernel.h

Go to the documentation of this file.

#ifndef CUDAGLOBALMASTERSERVERKERNEL_H
#define CUDAGLOBALMASTERSERVERKERNEL_H

#include "CudaGlobalMasterClient.h"
#include "CudaGlobalMasterServer.h"
#include "Lattice.h"

#if defined(NAMD_CUDA) && defined(NODEGROUP_FORCE_REGISTER)

void copyAtomsToClientsCUDA(
    bool copyPositions, bool copyMasses, bool copyCharges, bool copyTransforms,
    bool copyVelocities, const double *d_pos_x, const double *d_pos_y,
    const double *d_pos_z, const double *d_vel_x, const double *d_vel_y,
    const double *d_vel_z, const char3 *d_transform, const float *d_mass,
    const float *d_charge, const Lattice lat,
    const CudaGlobalMasterServer::CopyListTuple *d_copyList,
    size_t numCopyTuples, CudaGlobalMasterServer::ClientBuffer *d_clientBuffers,
    size_t numClients, cudaStream_t stream);

void copyAtomsToClientsCUDAMGPU(
    bool copyPositions, bool copyMasses, bool copyCharges, bool copyTransforms,
    bool copyVelocities, const double **d_peer_pos_x,
    const double **d_peer_pos_y, const double **d_peer_pos_z,
    const double **d_peer_vel_x, const double **d_peer_vel_y,
    const double **d_peer_vel_z, const char3 **d_peer_transform,
    const float **d_peer_mass, const float **d_peer_charge, const Lattice lat,
    const CudaGlobalMasterServer::CopyListTuple *d_copyList,
    size_t numCopyTuples, CudaGlobalMasterServer::ClientBuffer *d_clientBuffers,
    size_t numClients, cudaStream_t stream);

void copyTotalForcesToClientsCUDA(
    bool fixedOn, const double *d_f_normal_x, const double *d_f_normal_y,
    const double *d_f_normal_z, const double *d_f_nbond_x,
    const double *d_f_nbond_y, const double *d_f_nbond_z,
    const double *d_f_slow_x, const double *d_f_slow_y,
    const double *d_f_slow_z, const int *d_atomFixed,
    const CudaGlobalMasterServer::CopyListTuple *d_copyList,
    size_t numCopyTuples, CudaGlobalMasterServer::ClientBuffer *d_clientBuffers,
    size_t numClients, cudaStream_t stream);

void copyTotalForcesToClientsCUDAMGPU(
    bool fixedOn, const double **d_peer_f_normal_x,
    const double **d_peer_f_normal_y, const double **d_peer_f_normal_z,
    const double **d_peer_f_nbond_x, const double **d_peer_f_nbond_y,
    const double **d_peer_f_nbond_z, const double **d_peer_f_slow_x,
    const double **d_peer_f_slow_y, const double **d_peer_f_slow_z,
    const int **d_peer_atomFixed,
    const CudaGlobalMasterServer::CopyListTuple *d_copyList,
    size_t numCopyTuples, CudaGlobalMasterServer::ClientBuffer *d_clientBuffers,
    size_t numClients, cudaStream_t stream);

void addGlobalForcesFromClients(
    const int fixedOn, const int uniqueAtoms,
    double *d_f_global_x, double *d_f_global_y,
    double *d_f_global_z, const int *d_atomFixed,
    const CudaGlobalMasterServer::CopyListTuple *d_copyList,
    size_t numCopyTuples, CudaGlobalMasterServer::ClientBuffer *d_clientBuffers,
    size_t numClients, cudaStream_t stream);

void addGlobalForcesFromClientsMGPU(
    const int fixedOn, const int uniqueAtoms,
    double **d_peer_f_global_x, double **d_peer_f_global_y,
    double **d_peer_f_global_z, const int **d_peer_atomFixed,
    const CudaGlobalMasterServer::CopyListTuple *d_copyList,
    size_t numCopyTuples, CudaGlobalMasterServer::ClientBuffer *d_clientBuffers,
    size_t numClients, cudaStream_t stream);

#endif // defined(NAMD_CUDA) && defined(NODEGROUP_FORCE_REGISTER)

#endif // CUDAGLOBALMASTERSERVERKERNEL_H