NAMD
Public Member Functions | List of all members
CudaPmePencilXYZ Class Reference

#include <CudaPmeSolver.h>

Inheritance diagram for CudaPmePencilXYZ:

Public Member Functions

 CudaPmePencilXYZ ()
 
 CudaPmePencilXYZ (CkMigrateMessage *m)
 
void initialize (CudaPmeXYZInitMsg *msg)
 
void initializeDevice (InitDeviceMsg *msg)
 
void energyAndVirialDone ()
 

Detailed Description

Definition at line 56 of file CudaPmeSolver.h.

Constructor & Destructor Documentation

CudaPmePencilXYZ::CudaPmePencilXYZ ( )
inline

Definition at line 58 of file CudaPmeSolver.h.

58 {}
CudaPmePencilXYZ::CudaPmePencilXYZ ( CkMigrateMessage *  m)
inline

Definition at line 59 of file CudaPmeSolver.h.

59 {}

Member Function Documentation

void CudaPmePencilXYZ::energyAndVirialDone ( )

Definition at line 47 of file CudaPmeSolver.C.

47  {
48  submitReductions();
49  // deviceProxy[CkMyNode()].gatherForce();
50 }
void CudaPmePencilXYZ::initialize ( CudaPmeXYZInitMsg msg)

Definition at line 17 of file CudaPmeSolver.C.

References CudaPmeXYZInitMsg::pmeGrid.

17  {
18  pmeGrid = msg->pmeGrid;
19  delete msg;
20 }
void CudaPmePencilXYZ::initializeDevice ( InitDeviceMsg msg)

Definition at line 25 of file CudaPmeSolver.C.

References InitDeviceMsg::deviceProxy, ComputeNonbondedUtil::ewaldcof, Perm_cX_Y_Z, and stream.

25  {
26  // Store device proxy
27  deviceProxy = msg->deviceProxy;
28  delete msg;
29  int deviceID = deviceProxy.ckLocalBranch()->getDeviceID();
30  cudaStream_t stream = deviceProxy.ckLocalBranch()->getStream();
31  CProxy_ComputePmeCUDAMgr mgrProxy = deviceProxy.ckLocalBranch()->getMgrProxy();
32  // Setup fftCompute and pmeKSpaceCompute
33  fftCompute = new CudaFFTCompute(deviceID, stream);
34  pmeKSpaceCompute = new CudaPmeKSpaceCompute(pmeGrid, Perm_cX_Y_Z, 0, 0,
35  ComputeNonbondedUtil::ewaldcof, deviceID, stream);
36 }
CProxy_ComputePmeCUDADevice deviceProxy
Definition: CudaPmeSolver.h:44
__thread cudaStream_t stream

The documentation for this class was generated from the following files: