55int main(
int narg,
char *arg[]) {
56 Tpetra::ScopeGuard tscope(&narg, &arg);
57 Teuchos::RCP<const Teuchos::Comm<int> > comm = Tpetra::getDefaultComm();
59 int rank = comm->getRank();
60 int nprocs = comm->getSize();
61 int fail = 0, gfail = 0;
69 zgno_t myFirstId = rank * numLocalIds * numLocalIds;
71 for (
zlno_t i=0; i < numLocalIds; i++){
72 myIds[i] =
zgno_t(myFirstId+i);
74 weights[i*nWeights + 1] = (nprocs-rank) / (i+1);
78 std::vector<const zscalar_t *> weightValues;
79 std::vector<int> strides;
81 weightValues.push_back(
weights);
82 weightValues.push_back(
weights + 1);
87 weightValues, strides);
89 if (!
fail && ia.getLocalNumIDs() !=
size_t(numLocalIds)){
92 if (!
fail && ia.getNumWeightsPerID() != nWeights) {
98 int weightStridesIn[2];
100 ia.getIDsView(globalIdsIn);
102 for (
int w=0; !
fail && w < nWeights; w++) {
103 ia.getWeightsView(weightsIn[w], weightStridesIn[w], w);
108 int incr1 = weightStridesIn[0];
109 int incr2 = weightStridesIn[1];
111 for (
zlno_t i=0; !
fail && i < numLocalIds; i++){
112 if (globalIdsIn[i] !=
zgno_t(myFirstId+i)) {
115 if (!
fail && w1[i*incr1] != 1.0) {
118 if (!
fail && w2[i*incr2] !=
weights[i*nWeights+1]) {
131 std::cout <<
"PASS" << std::endl;