NAMD
ComputeGlobalMasterVirialCUDAKernel.h
Go to the documentation of this file.
1 #ifndef GLOBMTSVIRIALKERNEL_H
2 #define GLOBMTSVIRIALKERNEL_H
3 
4 #include "Lattice.h"
5 #include "CudaUtils.h"
6 #include "CudaRecord.h"
7 
8 #ifdef NODEGROUP_FORCE_REGISTER
9 void computeGlobalMasterVirial(
10  const int numPatches,
11  CudaLocalRecord* localRecords,
12  const double* __restrict d_pos_x,
13  const double* __restrict d_pos_y,
14  const double* __restrict d_pos_z,
15  const char3* __restrict d_transform,
16  double* __restrict f_global_x,
17  double* __restrict f_global_y,
18  double* __restrict f_global_z,
19  double3* __restrict d_extForce,
20  double3* __restrict h_extForce,
21  cudaTensor* __restrict d_virial,
22  cudaTensor* __restrict h_virial,
23  const Lattice lat,
24  unsigned int* __restrict tbcatomic,
25  cudaStream_t stream
26 );
27 
28 #endif // NODEGROUP_FORCE_REGISTER
29 #endif