ComputeGridForceCUDAKernel.h

Go to the documentation of this file.

#ifndef GRIDFORCEKERNEL_H
#define GRIDFORCEKERNEL_H

#include "Lattice.h"
#include "CudaUtils.h"
#include "CudaRecord.h"
#include "GridforceGridCUDAKernel.h"
#ifdef NODEGROUP_FORCE_REGISTER
void computeGridForce(
               const int doEnergy,
               const int doVirial,
               const GridforceGridCUDA& d_grid,
               const Lattice lat,
               const double* d_pos_x,
               const double* d_pos_y,
               const double* d_pos_z,
               const char3* d_transform,
               const int* gridForcedAtomIdxArr,
               const int* gridForcedAtomIdxMap,
               const float* d_gridded_charge,
               const float* d_gridded_scale,
               double* f_normal_x,
               double* f_normal_y,
               double* f_normal_z,
               double3* h_extForce,
               double3* d_extForce,
               double* h_extEnergy,
               double* d_extEnergy,
               cudaTensor* d_virial,
               cudaTensor* h_virial,
               const int numGriddedAtoms,
               unsigned int* d_tbcatomic,
               cudaStream_t stream

               );


#endif // NODEGROUP_FORCE_REGISTER
#endif // RESTRAINTSKERNEL_H