PmeSolverUtil.h File Reference

#include "PmeBase.h"
#include "NamdTypes.h"
#include "Lattice.h"

Go to the source code of this file.

Classes
struct	PatchInfo
class	FFTCompute
class	PmeKSpaceCompute
class	PmeRealSpaceCompute
class	PmeTranspose

Enumerations
enum	{ Perm_X_Y_Z, Perm_cX_Y_Z, Perm_Y_Z_cX, Perm_Z_cX_Y }

Functions
static int	getPencilIndexY (const PmeGrid &pmeGrid, const int y)
static int	getPencilIndexZ (const PmeGrid &pmeGrid, const int z)
static void	getPencilDim (const PmeGrid &pmeGrid, const int permutation, const int jblock, const int kblock, int &i0, int &i1, int &j0, int &j1, int &k0, int &k1)
static void	getBlockDim (const PmeGrid &pmeGrid, const int permutation, const int iblock, const int jblock, const int kblock, int &i0, int &i1, int &j0, int &j1, int &k0, int &k1)

Variables
const unsigned int	NUM_GRID_MAX = 5

Enumeration Type Documentation

anonymous enum

anonymous enum

Enumerator
Perm_X_Y_Z
Perm_cX_Y_Z
Perm_Y_Z_cX
Perm_Z_cX_Y

Definition at line 20 of file PmeSolverUtil.h.

{Perm_X_Y_Z, Perm_cX_Y_Z, Perm_Y_Z_cX, Perm_Z_cX_Y};

Function Documentation

getBlockDim()

static void getBlockDim ( const PmeGrid &  pmeGrid, const int  permutation, const int  iblock, const int  jblock, const int  kblock, int &  i0, int &  i1, int &  j0, int &  j1, int &  k0, int &  k1  )

static

Definition at line 89 of file PmeSolverUtil.h.

References getPencilDim(), NAMD_bug(), Perm_cX_Y_Z, Perm_X_Y_Z, Perm_Y_Z_cX, Perm_Z_cX_Y, PmeGrid::xBlocks, PmeGrid::yBlocks, and PmeGrid::zBlocks.

Referenced by CudaPmeTranspose::copyDataDeviceToDevice(), CudaPmeTranspose::copyDataHostToDevice(), CudaPmeTranspose::getBuffer(), PmePencilXY::initBlockSizes(), PmePencilX::initBlockSizes(), PmePencilY::initBlockSizes(), PmePencilZ::initBlockSizes(), and PmeTranspose::PmeTranspose().

                                                        {

  getPencilDim(pmeGrid, permutation, jblock, kblock, i0, i1, j0, j1, k0, k1);

  int iblocks;

  switch(permutation) {
    case Perm_X_Y_Z:
    iblocks = pmeGrid.xBlocks;
    break;
    case Perm_cX_Y_Z:
    iblocks = pmeGrid.xBlocks;
    break;
    case Perm_Y_Z_cX:
    iblocks = pmeGrid.yBlocks;
    break;
    case Perm_Z_cX_Y:
    iblocks = pmeGrid.zBlocks;
    break;
    default:
    NAMD_bug("getBlockDim, invalid permutation");
    break;
  }

  if (iblock < 0 || iblock >= iblocks)
    NAMD_bug("getBlockDim, invalid block index");

  int isize = i1-i0+1;

  i0 = isize*iblock/iblocks;
  i1 = isize*(iblock+1)/iblocks - 1;
}

getPencilDim()

static void getPencilDim ( const PmeGrid &  pmeGrid, const int  permutation, const int  jblock, const int  kblock, int &  i0, int &  i1, int &  j0, int &  j1, int &  k0, int &  k1  )

static

Definition at line 32 of file PmeSolverUtil.h.

References PmeGrid::K1, PmeGrid::K2, PmeGrid::K3, NAMD_bug(), Perm_cX_Y_Z, Perm_X_Y_Z, Perm_Y_Z_cX, Perm_Z_cX_Y, PmeGrid::xBlocks, PmeGrid::yBlocks, and PmeGrid::zBlocks.

Referenced by getBlockDim(), ComputePmeCUDAMgr::getHomePencil(), FFTCompute::init(), PmeKSpaceCompute::PmeKSpaceCompute(), PmeRealSpaceCompute::PmeRealSpaceCompute(), and PmeTranspose::PmeTranspose().

                                                        {

  int isize, jsize, ksize;
  int jblocks, kblocks;

  switch(permutation) {
    case Perm_X_Y_Z:
    isize = pmeGrid.K1;
    jsize = pmeGrid.K2;
    ksize = pmeGrid.K3;
    jblocks = pmeGrid.yBlocks;
    kblocks = pmeGrid.zBlocks;
    break;
    case Perm_cX_Y_Z:
    isize = pmeGrid.K1/2+1;
    jsize = pmeGrid.K2;
    ksize = pmeGrid.K3;
    jblocks = pmeGrid.yBlocks;
    kblocks = pmeGrid.zBlocks;
    break;
    case Perm_Y_Z_cX:
    isize = pmeGrid.K2;
    jsize = pmeGrid.K3;
    ksize = pmeGrid.K1/2+1;
    jblocks = pmeGrid.zBlocks;
    kblocks = pmeGrid.xBlocks;
    break;
    case Perm_Z_cX_Y:
    isize = pmeGrid.K3;
    jsize = pmeGrid.K1/2+1;
    ksize = pmeGrid.K2;
    jblocks = pmeGrid.xBlocks;
    kblocks = pmeGrid.yBlocks;
    break;
    default:
    NAMD_bug("getPencilDim, invalid permutation");
    break;
  }

  if (jblock < 0 || jblock >= jblocks || kblock < 0 || kblock >= kblocks)
    NAMD_bug("getPencilDim, invalid block indices");

  i0 = 0;
  i1 = isize - 1;

  j0 = jsize*jblock/jblocks;
  j1 = jsize*(jblock+1)/jblocks - 1;

  k0 = ksize*kblock/kblocks;
  k1 = ksize*(kblock+1)/kblocks - 1;
}

getPencilIndexY()

static int getPencilIndexY ( const PmeGrid &  pmeGrid, const int  y  )

inlinestatic

Definition at line 23 of file PmeSolverUtil.h.

References PmeGrid::K2, and PmeGrid::yBlocks.

Referenced by PmeAtomFiler::fileAtoms().

                                                                       {
  return (y*pmeGrid.yBlocks + pmeGrid.yBlocks - 1)/pmeGrid.K2;
}

getPencilIndexZ()

static int getPencilIndexZ ( const PmeGrid &  pmeGrid, const int  z  )

inlinestatic

Definition at line 28 of file PmeSolverUtil.h.

References PmeGrid::K3, and PmeGrid::zBlocks.

Referenced by PmeAtomFiler::fileAtoms().

                                                                       {
  return (z*pmeGrid.zBlocks + pmeGrid.zBlocks - 1)/pmeGrid.K3;
}

Variable Documentation

NUM_GRID_MAX

const unsigned int NUM_GRID_MAX = 5

Definition at line 9 of file PmeSolverUtil.h.

Referenced by ComputePmeCUDADevice::activate_pencils(), ComputePmeCUDAMgr::activate_pencils(), ComputePmeCUDADevice::ComputePmeCUDADevice(), DeviceBuffer::DeviceBuffer(), DeviceDataMsg::DeviceDataMsg(), ComputePmeCUDADevice::gatherForce(), ComputePmeCUDADevice::gatherForceDone(), ComputePmeCUDADevice::initialize(), CudaPmePencilXYZ::initializeDevice(), CudaPmePencilXY::initializeDevice(), CudaPmePencilX::initializeDevice(), CudaPmePencilY::initializeDevice(), CudaPmePencilZ::initializeDevice(), PmePencilX::PmePencilX(), PmePencilXY::PmePencilXY(), PmePencilXYZ::PmePencilXYZ(), PmePencilY::PmePencilY(), PmePencilZ::PmePencilZ(), PmeAtomStorage::setupAlch(), ComputePmeCUDADevice::spreadCharge(), PmePencilXYZ::submitReductions(), PmePencilZ::submitReductions(), ComputePmeCUDADevice::~ComputePmeCUDADevice(), PmePencilX::~PmePencilX(), PmePencilXY::~PmePencilXY(), PmePencilXYZ::~PmePencilXYZ(), PmePencilY::~PmePencilY(), and PmePencilZ::~PmePencilZ().