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 635 of file ComputePme.C.

Constructor & Destructor Documentation

NodePmeMgr()

NodePmeMgr::NodePmeMgr

(

)

Definition at line 686 of file ComputePme.C.

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

~NodePmeMgr()

NodePmeMgr::~NodePmeMgr

(

)

Definition at line 690 of file ComputePme.C.

                        {
  delete [] mgrObjects;
}

Member Function Documentation

initialize()

void NodePmeMgr::initialize

(

void

)

Definition at line 694 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 657 of file ComputePme.C.

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

recvTrans()

void NodePmeMgr::recvTrans

(

PmeTransMsg *

msg

)

Definition at line 703 of file ComputePme.C.

                                           {
  mgrObject->fwdSharedTrans(msg);
}

recvUngrid()

void NodePmeMgr::recvUngrid

(

PmeGridMsg *

msg

)

Definition at line 711 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 707 of file ComputePme.C.

                                               {
  mgrObject->fwdSharedUntrans(msg);
}

recvXTrans()

void NodePmeMgr::recvXTrans

(

PmeTransMsg *

msg

)

Definition at line 5070 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 5082 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 5092 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 5113 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 5102 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 719 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 725 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 731 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 4027 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 3800 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 638 of file ComputePme.C.

ComputePmeMgr

friend class ComputePmeMgr

friend

Definition at line 637 of file ComputePme.C.

Member Data Documentation

xm

CProxy_PmePencilMap NodePmeMgr::xm

Definition at line 660 of file ComputePme.C.

ym

CProxy_PmePencilMap NodePmeMgr::ym

Definition at line 661 of file ComputePme.C.

zm

CProxy_PmePencilMap NodePmeMgr::zm

Definition at line 662 of file ComputePme.C.

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

---

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

• ComputePme.C