CudaComputeNonbondedKernel.hip.h

Go to the documentation of this file.

#ifndef CUDACOMPUTENONBONDEDKERNEL_HIP_H
#define CUDACOMPUTENONBONDEDKERNEL_HIP_H
#include "CudaUtils.h"
#include "CudaRecord.h"
#include "CudaTileListKernel.hip.h"
#include "CudaNonbondedTables.h"
#include "HipDefines.h"
#if defined(NAMD_HIP)


struct AlchData{
  float scaling;
  float switchdist2;
  float cutoff2;
  float switchfactor;
  float alchVdwShiftCoeff;
  float alchLambda;
  float lambdaUp;
  float lambdaDown;
  float elecLambdaUp;
  float elecLambdaDown;
  float vdwLambdaUp;
  float vdwLambdaDown;

  float lambda2Up;
  float lambda2Down;
  float elecLambda2Up;
  float elecLambda2Down;
  float vdwLambda2Up;
  float vdwLambda2Down;

  float vdwShiftUp;
  float vdwShift2Up;
  float vdwShiftDown;
  float vdwShift2Down;
  bool  alchDecouple;
};

class CudaComputeNonbondedKernel {
private:

  const int deviceID;
  CudaNonbondedTables& cudaNonbondedTables;
  const bool doStreaming;

  // Exclusions
  // JM: Seems like nvcc makes this point to NULL by default but clang doesn't
  //     we rely on it being set to NULL so we can initialize it on setExclusiveByAtom() laters
  int2 *d_exclusionsByAtom = NULL; 
  unsigned int* overflowExclusions;
  size_t overflowExclusionsSize;

  int2* exclIndexMaxDiff;
  size_t exclIndexMaxDiffSize;

  // Atom indices
  int* atomIndex;
  size_t atomIndexSize;

  // VdW types
  int* vdwTypes;
  size_t vdwTypesSize;

  unsigned int* patchNumCount;
  size_t patchNumCountSize;

  int* patchReadyQueue;
  size_t patchReadyQueueSize;

  float *force_x, *force_y, *force_z, *force_w;
  size_t forceSize;
  float *forceSlow_x, *forceSlow_y, *forceSlow_z, *forceSlow_w;
  size_t forceSlowSize;

  // Drude/NbThole
  int *isDrude;
  size_t isDrudeSize;
  float* drudeAtomAlpha;
  size_t drudeAtomAlphaSize;
public:
  CudaComputeNonbondedKernel(int deviceID, CudaNonbondedTables& cudaNonbondedTables, bool doStreaming);
  ~CudaComputeNonbondedKernel();

  static __device__ __host__ __forceinline__ int 
  computeNumTiles(const int numAtoms, const int tilesize = BOUNDINGBOXSIZE) { 
    return (numAtoms+tilesize-1)/tilesize; 
  }

  static __device__ __host__ __forceinline__ int 
  computeAtomPad(const int numAtoms, const int tilesize = BOUNDINGBOXSIZE) { 
    return computeNumTiles(numAtoms, tilesize)*tilesize; 
  }

  void updateVdwTypesExcl(const int atomStorageSize, const int* h_vdwTypes,
    const int2* h_exclIndexMaxDiff, const int* h_atomIndex, cudaStream_t stream);

  void updateVdwTypesExclOnGPU(CudaTileListKernel& tlKernel,
    const int numPatches, const int atomStorageSize, const bool alchOn,
    CudaLocalRecord* localRecords,
    const int* d_vdwTypes, const int* d_id, const int* d_sortOrder, 
    const int* d_partition, cudaStream_t stream);

  void nonbondedForce(CudaTileListKernel& tlKernel,
  const int atomStorageSize, const bool atomsChanged, 
  const bool doMinimize, const bool doPairlist,
  const bool doEnergy, const bool doVirial, const bool doSlow, const bool doAlch, 
  const bool doAlchVdwForceSwitching,
  const bool doFEP, const bool doTI,
  const bool doNbThole, const bool doTable,
  const float3 lata, const float3 latb, const float3 latc,
  const float4* h_xyzq, const float cutoff2,
  const CudaNBConstants nbConstants,  
  float4* d_forces, float4* d_forcesSlow,
  float4* h_forces, float4* h_forcesSlow, AlchData *srcFlags, 
  bool lambdaWindowUpdated, char *part,
  bool CUDASOAintegratorOn, bool useDeviceMigration,  
  const float drudeNbtholeCut2,
  cudaStream_t stream);

  void reduceVirialEnergy(CudaTileListKernel& tlKernel,
    const int atomStorageSize, const bool doEnergy, const bool doVirial, const bool doSlow, const bool doGBIS,
    float4* d_forces, float4* d_forcesSlow,
    VirialEnergy* d_virialEnergy, cudaStream_t stream);

  void getVirialEnergy(VirialEnergy* h_virialEnergy, cudaStream_t stream);

  void bindExclusions(int numExclusions, unsigned int* exclusion_bits);

  int* getPatchReadyQueue();

  void reallocate_forceSOA(int atomStorageSize); 

  void setExclusionsByAtom(int2* h_data, const int num_atoms);

  void updateDrudeData(const int atomStorageSize, const float* h_drudeAtomAlpha, const int* h_isDrude, cudaStream_t stream);
};

#endif // NAMD_HIP
#endif // CUDACOMPUTENONBONDEDKERNEL_H