Lines Matching refs:pj
716 PetscMPIInt **procs, pi, pj, pim, pip, pjm, pjp, PIi, PJi; in DMDAGetNeighborsRank() local
731 for (pj = 0; pj < PJ; pj++) { in DMDAGetNeighborsRank()
732 PetscCall(PetscMalloc1(PI, &procs[pj])); in DMDAGetNeighborsRank()
734 procs[pj][pi] = rank; in DMDAGetNeighborsRank()
740 pj = neighbors[0] / PI; in DMDAGetNeighborsRank()
744 pjm = pj - 1; in DMDAGetNeighborsRank()
746 pjp = (pj + 1) % PJi; in DMDAGetNeighborsRank()
748 neighbors[1] = procs[pj][pim]; in DMDAGetNeighborsRank()
752 neighbors[5] = procs[pj][pip]; in DMDAGetNeighborsRank()
763 if (pj == 0) neighbors[6] = neighbors[7] = neighbors[8] = neighbors[0]; in DMDAGetNeighborsRank()
764 if (pj == PJ - 1) neighbors[2] = neighbors[3] = neighbors[4] = neighbors[0]; in DMDAGetNeighborsRank()
767 for (pj = 0; pj < PJ; pj++) PetscCall(PetscFree(procs[pj])); in DMDAGetNeighborsRank()