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 (unsigned int iGrid)

Detailed Description

Definition at line 57 of file CudaPmeSolver.h.

Constructor & Destructor Documentation

CudaPmePencilXYZ() [1/2]

CudaPmePencilXYZ::CudaPmePencilXYZ ( )

inline

Definition at line 59 of file CudaPmeSolver.h.

{}

CudaPmePencilXYZ() [2/2]

CudaPmePencilXYZ::CudaPmePencilXYZ ( CkMigrateMessage *  m )

inline

Definition at line 60 of file CudaPmeSolver.h.

{}

Member Function Documentation

energyAndVirialDone()

void CudaPmePencilXYZ::energyAndVirialDone

(

unsigned int

iGrid

)

Definition at line 61 of file CudaPmeSolver.C.

                                                             {
  submitReductions(iGrid);
  // deviceProxy[CkMyNode()].gatherForce();
}

initialize()

void CudaPmePencilXYZ::initialize

(

CudaPmeXYZInitMsg *

msg

)

Definition at line 18 of file CudaPmeSolver.C.

References CudaPmeXYZInitMsg::pmeGrid.

                                                        {
  pmeGrid = msg->pmeGrid;
  delete msg;
}

initializeDevice()

void CudaPmePencilXYZ::initializeDevice

(

InitDeviceMsg *

msg

)

Definition at line 26 of file CudaPmeSolver.C.

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

                                                          {
  // Store device proxy
  deviceProxy = msg->deviceProxy;
  delete msg;
  int deviceID = deviceProxy.ckLocalBranch()->getDeviceID();
  cudaStream_t stream = deviceProxy.ckLocalBranch()->getStream();
  CProxy_ComputePmeCUDAMgr mgrProxy = deviceProxy.ckLocalBranch()->getMgrProxy();
  // Setup fftCompute and pmeKSpaceCompute
  for (unsigned int iGrid = 0; iGrid < NUM_GRID_MAX; ++iGrid) {
    if (deviceProxy.ckLocalBranch()->isGridEnabled(iGrid) == true) {
      fftComputes[iGrid] = new CudaFFTCompute(deviceID, stream);
      pmeKSpaceComputes[iGrid] = new CudaPmeKSpaceCompute(pmeGrid, Perm_cX_Y_Z, 0, 0, ComputeNonbondedUtil::ewaldcof, deviceID, stream, iGrid);
      energyReady[iGrid] = 0;
    } else {
      fftComputes[iGrid] = NULL;
      pmeKSpaceComputes[iGrid] = NULL;
      energyReady[iGrid] = -1;
    }
  }
}

---

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

• CudaPmeSolver.h
• CudaPmeSolver.C