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

Constructor & Destructor Documentation

NodePmeMgr::NodePmeMgr ( )

Definition at line 664 of file ComputePme.C.

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

NodePmeMgr::~NodePmeMgr ( )

Definition at line 668 of file ComputePme.C.

                         {
   delete [] mgrObjects;
 }

Member Function Documentation

void NodePmeMgr::initialize ( void )

Definition at line 672 of file ComputePme.C.

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

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

inline

Definition at line 635 of file ComputePme.C.

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

void NodePmeMgr::recvTrans ( PmeTransMsg * msg )

Definition at line 681 of file ComputePme.C.

                                            {
   mgrObject->fwdSharedTrans(msg);
 }

void NodePmeMgr::recvUngrid ( PmeGridMsg * msg )

Definition at line 689 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
 }

void NodePmeMgr::recvUntrans ( PmeUntransMsg * msg )

Definition at line 685 of file ComputePme.C.

                                                {
   mgrObject->fwdSharedUntrans(msg);
 }

void NodePmeMgr::recvXTrans ( PmeTransMsg * msg )

Definition at line 4987 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;
 }

void NodePmeMgr::recvYTrans ( PmeTransMsg * msg )

Definition at line 4999 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;
  }

void NodePmeMgr::recvYUntrans ( PmeUntransMsg * msg )

Definition at line 5009 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;
  }

void NodePmeMgr::recvZGrid ( PmeGridMsg * msg )

Definition at line 5030 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
 }

void NodePmeMgr::recvZUntrans ( PmeUntransMsg * msg )

Definition at line 5019 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;
 }

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

Definition at line 697 of file ComputePme.C.

References ComputePmeMgr::fftw_plan_lock.

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

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

Definition at line 703 of file ComputePme.C.

References ComputePmeMgr::fftw_plan_lock.

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

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

Definition at line 709 of file ComputePme.C.

References ComputePmeMgr::fftw_plan_lock.

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

void NodePmeMgr::sendDataHelper ( int iter )

Definition at line 3955 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
 }

void NodePmeMgr::sendPencilsHelper ( int iter )

Definition at line 3728 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
 }