ComputeConsForceCUDAKernel.h

Go to the documentation of this file.

#ifndef CONSFORCEKERNEL_H
#define CONSFORCEKERNEL_H

#include "Lattice.h"
#include "CudaUtils.h"
#include "CudaRecord.h"

#ifdef NODEGROUP_FORCE_REGISTER
void computeConsForce(
  const bool doVirial,
  const int nConsForceAtoms,
  const int*    __restrict d_consForceSOA,
  const int*    __restrict d_consForceID,
  const double* __restrict d_pos_x,
  const double* __restrict d_pos_y,
  const double* __restrict d_pos_z,
  const double3* __restrict d_consForce,
  const char3* __restrict d_transform,
  double*       __restrict f_normal_x,
  double*       __restrict f_normal_y,
  double*       __restrict f_normal_z,
  double3*      __restrict d_extForce,
  double3*      __restrict h_extForce,
  cudaTensor*   __restrict d_virial,
  cudaTensor*   __restrict h_virial,
  const double scaling,
  const Lattice            lat,
  unsigned int* __restrict tbcatomic,
  cudaStream_t stream
);

#endif // NODEGROUP_FORCE_REGISTER
#endif