NAMD
ComputeGroupRes1GroupCUDAKernel.h
Go to the documentation of this file.
1 #ifndef GROUP_RESTRAINTS_1GROUP_CUDA_KERNEL_H
2 #define GROUP_RESTRAINTS_1GROUP_CUDA_KERNEL_H
3 
4 #include "Lattice.h"
5 #include "CudaUtils.h"
6 #include "CudaRecord.h"
7 
8 #ifdef NODEGROUP_FORCE_REGISTER
9 
14 void computeGroupRestraint_1Group(
15  const int useMagnitude,
16  const int doEnergy,
17  const int doVirial,
18  const int numRestrainedGroup,
19  const int restraintExp,
20  const double restraintK,
21  const double3 resCenterVec,
22  const double3 resDirection,
23  const double inv_group2_mass,
24  const int* d_groupAtomsSOAIndex,
25  const Lattice &lat,
26  const char3* d_transform,
27  const float* d_mass,
28  const double* d_pos_x,
29  const double* d_pos_y,
30  const double* d_pos_z,
31  double* d_f_normal_x,
32  double* d_f_normal_y,
33  double* d_f_normal_z,
34  cudaTensor* d_virial,
35  cudaTensor* h_extVirial,
36  double* h_resEnergy,
37  double3* h_resForce,
38  double3* h_group1COMRef,
39  double3* h_group2COM,
40  double3* h_diffCOM,
41  double3* d_group2COM,
42  unsigned int* d_tbcatomic,
43  cudaStream_t stream);
44 
45 #endif // NODEGROUP_FORCE_REGISTER
46 #endif // GROUP_RESTRAINTS_1GROUP_CUDA_KERNEL_H
Generated by   doxygen 1.8.14