Revision e07333e63a376a103d320cc43c22df0b78b7f2a8 authored by Tim Haines on 30 October 2018, 03:15:40 UTC, committed by Tim Haines on 11 November 2018, 10:27:44 UTC
This requires https://charm.cs.illinois.edu/gerrit/#/c/cosmo/changa/+/3393/ https://charm.cs.illinois.edu/gerrit/#/c/cosmo/changa/+/3602/ https://charm.cs.illinois.edu/gerrit/#/c/charm/+/4753/ https://charm.cs.illinois.edu/gerrit/#/c/charm/+/4755/ Change-Id: I4d0adea6284941a2d1a62ace98faaeea39a73b3c
1 parent 2fbebec
Orb3dLB.h
/**
* \addtogroup CkLdb
*/
/*@{*/
#ifndef _ORB3DLB_H_
#define _ORB3DLB_H_
#include "CentralLB.h"
#include "MapStructures.h"
#include "Orb3dLB.decl.h"
#include "TaggedVector3D.h"
#include <queue>
void CreateOrb3dLB();
BaseLB * AllocateOrb3dLB();
class Orb3dLB : public CBase_Orb3dLB {
friend class MultistepLB;
private:
bool firstRound;
bool centroidsAllocated;
ComparatorFn compares[NDIMS];
ComparatorFn pc[NDIMS];
// pointer to stats->to_proc
CkVec<int> *mapping;
int procsPerNode;
CkVec<TPObject> tps;
CkVec<Node> nodes;
void init();
bool QueryBalanceNow(int step);
void printData(BaseLB::LDStats &stats, int phase, int *revObjMap);
public:
Orb3dLB(const CkLBOptions &);
Orb3dLB(CkMigrateMessage *m) : CBase_Orb3dLB(m) {init();}
void work(BaseLB::LDStats* stats);
void directMap(int tpstart, int tpend, int nodestart, int nodeend);
void map(int tpstart, int tpend, int nodestart, int nodeend, int xs, int ys, int zs, int dim);
int nextDim(int dim, int xs, int ys, int zs);
void partitionEvenLoad(int tpstart, int tpend, int &tpmid);
void halveNodes(int nodestart, int nodeend, int &nodemid);
};
#endif /* _ORB3DLB_H_ */
/*@}*/
Computing file changes ...