PmeTranspose Class Reference

#include <PmeSolverUtil.h>

Inheritance diagram for PmeTranspose:

Public Member Functions
	PmeTranspose (PmeGrid pmeGrid, const int permutation, const int jblock, const int kblock)
virtual	~PmeTranspose ()
virtual void	transposeXYZtoYZX (const float2 *data)=0
virtual void	transposeXYZtoZXY (const float2 *data)=0

Protected Attributes
PmeGrid	pmeGrid
const int	permutation
const int	jblock
const int	kblock
int	isize
int	jsize
int	ksize
int	dataSize
int	nblock
std::vector< int >	pos

Detailed Description

Definition at line 411 of file PmeSolverUtil.h.

Constructor & Destructor Documentation

PmeTranspose::PmeTranspose ( PmeGrid  pmeGrid, const int  permutation, const int  jblock, const int  kblock  )

inline

Definition at line 421 of file PmeSolverUtil.h.

References dataSize, getBlockDim(), getPencilDim(), isize, jsize, ksize, NAMD_bug(), nblock, Perm_cX_Y_Z, Perm_Y_Z_cX, Perm_Z_cX_Y, pos, PmeGrid::xBlocks, PmeGrid::yBlocks, and PmeGrid::zBlocks.

                                                                                           : 
    pmeGrid(pmeGrid), permutation(permutation), jblock(jblock), kblock(kblock) {

    int i0, i1, j0, j1, k0, k1;
    getPencilDim(pmeGrid, permutation, jblock, kblock, i0, i1, j0, j1, k0, k1);
    isize = (i1-i0+1);
    jsize = (j1-j0+1);
    ksize = (k1-k0+1);
    dataSize = (i1-i0+1)*(j1-j0+1)*(k1-k0+1);

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

    // Pos marks the beginning of blocks
    pos.resize(nblock+1);

    int x1;
    for (int iblock=0;iblock < nblock;iblock++) {
      // Get the dimension of the transpose block
      // We are transposing nblock of these blocks and storing them into a pencil of size
      // ny * nz * xsize at location ny*nz*x0
      int x0, y0dummy, y1dummy, z0dummy, z1dummy;
      getBlockDim(pmeGrid, permutation, iblock, jblock, kblock, x0, x1, y0dummy, y1dummy, z0dummy, z1dummy);

      pos[iblock] = x0;
    }
    // Last position begins at x1+1
    pos[nblock] = x1+1;
  }

virtual PmeTranspose::~PmeTranspose ( )

inlinevirtual

Definition at line 462 of file PmeSolverUtil.h.

{}

Member Function Documentation

virtual void PmeTranspose::transposeXYZtoYZX ( const float2 *  data )

pure virtual

Implemented in CudaPmeTranspose.

virtual void PmeTranspose::transposeXYZtoZXY ( const float2 *  data )

pure virtual

Implemented in CudaPmeTranspose.

Member Data Documentation

int PmeTranspose::dataSize

protected

Definition at line 417 of file PmeSolverUtil.h.

Referenced by CudaPmeTranspose::copyDataDeviceToHost(), CudaPmeTranspose::CudaPmeTranspose(), and PmeTranspose().

int PmeTranspose::isize

protected

Definition at line 416 of file PmeSolverUtil.h.

Referenced by CudaPmeTranspose::copyDataDeviceToDevice(), CudaPmeTranspose::copyDataHostToDevice(), PmeTranspose(), CudaPmeTranspose::transposeXYZtoYZX(), and CudaPmeTranspose::transposeXYZtoZXY().

const int PmeTranspose::jblock

protected

Definition at line 415 of file PmeSolverUtil.h.

Referenced by CudaPmeTranspose::copyDataDeviceToDevice(), CudaPmeTranspose::copyDataHostToDevice(), CudaPmeTranspose::copyDataToPeerDeviceYZX(), CudaPmeTranspose::copyDataToPeerDeviceZXY(), and CudaPmeTranspose::getBuffer().

int PmeTranspose::jsize

protected

Definition at line 416 of file PmeSolverUtil.h.

Referenced by CudaPmeTranspose::copyDataDeviceToDevice(), CudaPmeTranspose::copyDataDeviceToHost(), CudaPmeTranspose::copyDataHostToDevice(), PmeTranspose(), CudaPmeTranspose::setDataPtrsYZX(), CudaPmeTranspose::setDataPtrsZXY(), CudaPmeTranspose::transposeXYZtoYZX(), and CudaPmeTranspose::transposeXYZtoZXY().

const int PmeTranspose::kblock

protected

Definition at line 415 of file PmeSolverUtil.h.

Referenced by CudaPmeTranspose::copyDataDeviceToDevice(), CudaPmeTranspose::copyDataHostToDevice(), CudaPmeTranspose::copyDataToPeerDeviceYZX(), CudaPmeTranspose::copyDataToPeerDeviceZXY(), and CudaPmeTranspose::getBuffer().

int PmeTranspose::ksize

protected

Definition at line 416 of file PmeSolverUtil.h.

Referenced by CudaPmeTranspose::copyDataDeviceToHost(), PmeTranspose(), CudaPmeTranspose::setDataPtrsYZX(), CudaPmeTranspose::setDataPtrsZXY(), CudaPmeTranspose::transposeXYZtoYZX(), and CudaPmeTranspose::transposeXYZtoZXY().

int PmeTranspose::nblock

protected

Definition at line 418 of file PmeSolverUtil.h.

Referenced by CudaPmeTranspose::copyDataDeviceToDevice(), CudaPmeTranspose::copyDataDeviceToHost(), CudaPmeTranspose::copyDataHostToDevice(), CudaPmeTranspose::CudaPmeTranspose(), CudaPmeTranspose::getBuffer(), PmeTranspose(), CudaPmeTranspose::setDataPtrsYZX(), CudaPmeTranspose::setDataPtrsZXY(), CudaPmeTranspose::transposeXYZtoYZX(), and CudaPmeTranspose::transposeXYZtoZXY().

const int PmeTranspose::permutation

protected

Definition at line 414 of file PmeSolverUtil.h.

Referenced by CudaPmeTranspose::copyDataDeviceToDevice(), CudaPmeTranspose::copyDataHostToDevice(), CudaPmeTranspose::getBuffer(), CudaPmeTranspose::transposeXYZtoYZX(), and CudaPmeTranspose::transposeXYZtoZXY().

PmeGrid PmeTranspose::pmeGrid

protected

Definition at line 413 of file PmeSolverUtil.h.

Referenced by CudaPmeTranspose::copyDataDeviceToDevice(), CudaPmeTranspose::copyDataHostToDevice(), CudaPmeTranspose::getBuffer(), CudaPmeTranspose::setDataPtrsYZX(), and CudaPmeTranspose::setDataPtrsZXY().

std::vector<int> PmeTranspose::pos

protected

Definition at line 419 of file PmeSolverUtil.h.

Referenced by CudaPmeTranspose::copyDataDeviceToHost(), PmeTranspose(), CudaPmeTranspose::setDataPtrsYZX(), and CudaPmeTranspose::setDataPtrsZXY().

---

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

• PmeSolverUtil.h