NodePmeMgr Class Reference

Inheritance diagram for NodePmeMgr:

Public Member Functions
	NodePmeMgr ()
	~NodePmeMgr ()
void	initialize ()
void	sendDataHelper (int)
void	sendPencilsHelper (int)
void	recvTrans (PmeTransMsg *)
void	recvUntrans (PmeUntransMsg *)
void	registerXPencil (CkArrayIndex3D, PmeXPencil *)
void	registerYPencil (CkArrayIndex3D, PmeYPencil *)
void	registerZPencil (CkArrayIndex3D, PmeZPencil *)
void	recvXTrans (PmeTransMsg *)
void	recvYTrans (PmeTransMsg *)
void	recvYUntrans (PmeUntransMsg *)
void	recvZGrid (PmeGridMsg *)
void	recvZUntrans (PmeUntransMsg *)
void	recvUngrid (PmeGridMsg *)
void	recvPencilMapProxies (CProxy_PmePencilMap _xm, CProxy_PmePencilMap _ym, CProxy_PmePencilMap _zm)

Public Attributes
CProxy_PmePencilMap	xm
CProxy_PmePencilMap	ym
CProxy_PmePencilMap	zm

Friends
class	ComputePmeMgr
class	ComputePme

Detailed Description

Definition at line 633 of file ComputePme.C.

Constructor & Destructor Documentation

NodePmeMgr()

NodePmeMgr::NodePmeMgr

(

)

Definition at line 684 of file ComputePme.C.

                       {
  mgrObjects = new ComputePmeMgr*[CkMyNodeSize()];
}

~NodePmeMgr()

NodePmeMgr::~NodePmeMgr

(

)

Definition at line 688 of file ComputePme.C.

                        {
  delete [] mgrObjects;
}

Member Function Documentation

initialize()

void NodePmeMgr::initialize

(

void

)

Definition at line 692 of file ComputePme.C.

                            {
  CProxy_ComputePmeMgr proxy = CkpvAccess(BOCclass_group).computePmeMgr;
  mgrObjects[CkMyRank()] = proxy.ckLocalBranch();
  if ( CkMyRank() == 0 ) {
    mgrProxy = proxy;
    mgrObject = proxy.ckLocalBranch();
  }
}

recvPencilMapProxies()

void NodePmeMgr::recvPencilMapProxies ( CProxy_PmePencilMap  _xm, CProxy_PmePencilMap  _ym, CProxy_PmePencilMap  _zm  )

inline

Definition at line 655 of file ComputePme.C.

                                                                                                      {
      xm=_xm; ym=_ym; zm=_zm;
  }

recvTrans()

void NodePmeMgr::recvTrans

(

PmeTransMsg *

msg

)

Definition at line 701 of file ComputePme.C.

                                           {
  mgrObject->fwdSharedTrans(msg);
}

recvUngrid()

void NodePmeMgr::recvUngrid

(

PmeGridMsg *

msg

)

Definition at line 709 of file ComputePme.C.

References NAMD_bug().

                                           {
#if defined(NAMD_CUDA) || defined(NAMD_HIP)
  masterPmeMgr->recvUngrid(msg);
#else
  NAMD_bug("NodePmeMgr::recvUngrid called in non-CUDA build.");
#endif
}

recvUntrans()

void NodePmeMgr::recvUntrans

(

PmeUntransMsg *

msg

)

Definition at line 705 of file ComputePme.C.

                                               {
  mgrObject->fwdSharedUntrans(msg);
}

recvXTrans()

void NodePmeMgr::recvXTrans

(

PmeTransMsg *

msg

)

Definition at line 5012 of file ComputePme.C.

References PmeTransMsg::destElem, and PmeXPencil::node_process_trans().

                                            {
  //  CkPrintf("[%d] NodePmeMgr recvXTrans for %d %d %d\n",CkMyPe(),msg->destElem.index[0],msg->destElem.index[1],msg->destElem.index[2]);
  PmeXPencil *target=xPencilObj.get(msg->destElem);
#if DEBUG_NODE_PAR_RECV
  if(target == NULL)
    CkAbort("xpencil in recvXTrans not found, debug registeration");
#endif  
    target->node_process_trans(msg);
  delete msg;
}

recvYTrans()

void NodePmeMgr::recvYTrans

(

PmeTransMsg *

msg

)

Definition at line 5024 of file ComputePme.C.

References PmeTransMsg::destElem, and PmeYPencil::node_process_trans().

                                            {
  //  CkPrintf("[%d] NodePmeMgr recvYTrans for %d %d %d\n",CkMyPe(),msg->destElem.index[0],msg->destElem.index[1],msg->destElem.index[2]);
  PmeYPencil *target=yPencilObj.get(msg->destElem);
#if DEBUG_NODE_PAR_RECV
  if(target == NULL)
    CkAbort("ypencil in recvYTrans not found, debug registeration");
#endif  
    target->node_process_trans(msg);
  delete msg;
 }

recvYUntrans()

void NodePmeMgr::recvYUntrans

(

PmeUntransMsg *

msg

)

Definition at line 5034 of file ComputePme.C.

References PmeUntransMsg::destElem, and PmeYPencil::node_process_untrans().

                                                {
  //  CkPrintf("[%d] NodePmeMgr recvYUntrans for %d %d %d\n",CkMyPe(),msg->destElem.index[0],msg->destElem.index[1],msg->destElem.index[2]);
  PmeYPencil *target=yPencilObj.get(msg->destElem);
#if DEBUG_NODE_PAR_RECV  
  if(target == NULL)
    CkAbort("ypencil in recvYUntrans not found, debug registeration");
#endif  
    target->node_process_untrans(msg);
  delete msg;
 }

recvZGrid()

void NodePmeMgr::recvZGrid

(

PmeGridMsg *

msg

)

Definition at line 5055 of file ComputePme.C.

References PmeGridMsg::destElem, and PmeZPencil::node_process_grid().

                                          {
  //CkPrintf("[%d] NodePmeMgr %p recvGrid for %d %d %d\n",CkMyPe(),this,msg->destElem.index[0],msg->destElem.index[1],msg->destElem.index[2]);
  PmeZPencil *target=zPencilObj.get(msg->destElem);
#if DEBUG_NODE_PAR_RECV
  if(target == NULL){
    CkAbort("zpencil in recvZGrid not found, debug registeration");
  }
#endif
  target->node_process_grid(msg); //msg is stored inside node_proces_grid
}

recvZUntrans()

void NodePmeMgr::recvZUntrans

(

PmeUntransMsg *

msg

)

Definition at line 5044 of file ComputePme.C.

References PmeUntransMsg::destElem, and PmeZPencil::node_process_untrans().

                                                {
  //CkPrintf("[%d] NodePmeMgr recvZUntrans for %d %d %d\n",CkMyPe(),msg->destElem.index[0],msg->destElem.index[1],msg->destElem.index[2]);
  PmeZPencil *target=zPencilObj.get(msg->destElem);
#if DEBUG_NODE_PAR_RECV
  if(target == NULL)
    CkAbort("zpencil in recvZUntrans not found, debug registeration");
#endif
  target->node_process_untrans(msg);
  delete msg;
}

registerXPencil()

void NodePmeMgr::registerXPencil	(	CkArrayIndex3D	idx,
		PmeXPencil *	obj
	)

Definition at line 717 of file ComputePme.C.

References ComputePmeMgr::fftw_plan_lock.

{
  CmiLock(ComputePmeMgr::fftw_plan_lock);
  xPencilObj.put(idx)=obj;
  CmiUnlock(ComputePmeMgr::fftw_plan_lock);
}

registerYPencil()

void NodePmeMgr::registerYPencil	(	CkArrayIndex3D	idx,
		PmeYPencil *	obj
	)

Definition at line 723 of file ComputePme.C.

References ComputePmeMgr::fftw_plan_lock.

{
  CmiLock(ComputePmeMgr::fftw_plan_lock);
  yPencilObj.put(idx)=obj;
  CmiUnlock(ComputePmeMgr::fftw_plan_lock);
}

registerZPencil()

void NodePmeMgr::registerZPencil	(	CkArrayIndex3D	idx,
		PmeZPencil *	obj
	)

Definition at line 729 of file ComputePme.C.

References ComputePmeMgr::fftw_plan_lock.

{
  CmiLock(ComputePmeMgr::fftw_plan_lock);
  zPencilObj.put(idx)=obj;
  CmiUnlock(ComputePmeMgr::fftw_plan_lock);
}

sendDataHelper()

void NodePmeMgr::sendDataHelper

(

int

iter

)

Definition at line 3980 of file ComputePme.C.

References NAMD_bug(), ComputePmeMgr::sendDataHelper_errors, ComputePmeMgr::sendDataHelper_lattice, ComputePmeMgr::sendDataHelper_sequence, ComputePmeMgr::sendDataHelper_sourcepe, and ComputePmeMgr::sendDataPart().

Referenced by ComputePmeMgr::sendDataHelper().

                                        {
#if defined(NAMD_CUDA) || defined(NAMD_HIP)
  ComputePmeMgr *obj = masterPmeMgr;
  obj->sendDataPart(iter, iter, *obj->sendDataHelper_lattice, obj->sendDataHelper_sequence, obj->sendDataHelper_sourcepe, obj->sendDataHelper_errors);
#else
  NAMD_bug("NodePmeMgr::sendDataHelper called in non-CUDA build");
#endif
}

sendPencilsHelper()

void NodePmeMgr::sendPencilsHelper

(

int

iter

)

Definition at line 3753 of file ComputePme.C.

References NAMD_bug(), ComputePmeMgr::sendDataHelper_lattice, ComputePmeMgr::sendDataHelper_sequence, ComputePmeMgr::sendDataHelper_sourcepe, and ComputePmeMgr::sendPencilsPart().

Referenced by ComputePmeMgr::sendPencilsHelper().

                                           {
#if defined(NAMD_CUDA) || defined(NAMD_HIP)
  ComputePmeMgr *obj = masterPmeMgr;
  obj->sendPencilsPart(iter, iter, *obj->sendDataHelper_lattice, obj->sendDataHelper_sequence, obj->sendDataHelper_sourcepe);
#else
  NAMD_bug("NodePmeMgr::sendPencilsHelper called in non-CUDA build");
#endif
}

Friends And Related Function Documentation

ComputePme

friend class ComputePme

friend

Definition at line 636 of file ComputePme.C.

ComputePmeMgr

friend class ComputePmeMgr

friend

Definition at line 635 of file ComputePme.C.

Member Data Documentation

xm

CProxy_PmePencilMap NodePmeMgr::xm

Definition at line 658 of file ComputePme.C.

ym

CProxy_PmePencilMap NodePmeMgr::ym

Definition at line 659 of file ComputePme.C.

zm

CProxy_PmePencilMap NodePmeMgr::zm

Definition at line 660 of file ComputePme.C.

Referenced by ComputePmeMgr::sendPencils(), and ComputePmeMgr::sendPencilsPart().

---

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

• ComputePme.C